From 0dad501fbd321b6aed8ecf7c84528f8e41f730dc Mon Sep 17 00:00:00 2001 From: saltacc Date: Wed, 13 Dec 2023 22:15:46 -0800 Subject: [PATCH 1/2] fix model not existing --- animation/rerender-lora/rerender.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/animation/rerender-lora/rerender.py b/animation/rerender-lora/rerender.py index 4b72fa0..3ca563a 100644 --- a/animation/rerender-lora/rerender.py +++ b/animation/rerender-lora/rerender.py @@ -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: @@ -340,7 +342,7 @@ def rerender(cfg: RerenderConfig, first_img_only: bool, key_video_path: str): masks = [] for j in range(ddim_steps): if j <= ddim_steps * mask_period[ - 0] or j >= ddim_steps * mask_period[1]: +0] or j >= ddim_steps * mask_period[1]: masks += [None] else: masks += [mask * cfg.mask_strength] From 65dd822805378fad7a9463897cc6655bf32c9f8f Mon Sep 17 00:00:00 2001 From: saltacc Date: Wed, 13 Dec 2023 22:19:34 -0800 Subject: [PATCH 2/2] whoops --- animation/rerender-lora/rerender.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/animation/rerender-lora/rerender.py b/animation/rerender-lora/rerender.py index 3ca563a..32f8eb8 100644 --- a/animation/rerender-lora/rerender.py +++ b/animation/rerender-lora/rerender.py @@ -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]