storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-09 00:09:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

whoops

Browse Source
This commit is contained in:
saltacc
2023-12-13 22:19:34 -08:00
parent 0dad501fbd
commit 65dd822805
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animation/rerender-lora/rerender.py
+1 -2
View File
@@ -341,8 +341,7 @@ def rerender(cfg: RerenderConfig, first_img_only: bool, key_video_path: str):
noise_rescale = torch.ones_like(mask)
masks = []
for j in range(ddim_steps):
if j <= ddim_steps * mask_period[
0] or j >= ddim_steps * mask_period[1]:
if j <= ddim_steps * mask_period[0] or j >= ddim_steps * mask_period[1]:
masks += [None]
else:
masks += [mask * cfg.mask_strength]