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

Merge pull request #27 from storytold/rerender

Browse Source 
fix model not existing
This commit is contained in:
ArEnSc
2023-12-14 10:10:35 -05:00
committed by GitHub
parent 8434144ded 65dd822805
commit 2b05b454df
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animation/rerender-lora/rerender.py
+3 -2
View File
@@ -92,6 +92,8 @@ def rerender(cfg: RerenderConfig, first_img_only: bool, key_video_path: str):
elif model_ext == '.ckpt' or model_ext == '.pth':
model.load_state_dict(torch.load(cfg.sd_model)['state_dict'],
strict=False)
else:
raise Exception(f'Unknown model extension {model_ext}')
# apply lora if exists
if cfg.lora_path:
@@ -339,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]