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

fix model not existing

Browse Source
This commit is contained in:
saltacc
2023-12-13 22:15:46 -08:00
parent c926010fda
commit 0dad501fbd
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animation/rerender-lora/rerender.py
+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': elif model_ext == '.ckpt' or model_ext == '.pth':
model.load_state_dict(torch.load(cfg.sd_model)['state_dict'], model.load_state_dict(torch.load(cfg.sd_model)['state_dict'],
strict=False) strict=False)
else:
raise Exception(f'Unknown model extension {model_ext}')
# apply lora if exists # apply lora if exists
if cfg.lora_path: if cfg.lora_path: