modify

app.py

@@ -160,9 +160,6 @@ diffusion_model = diffusion_model.to('cuda')
 # 加载模型检查点
 # ckpt_path = 'models/relvid_mm_sd15_fbc_unet.pth' #! change
 # ckpt_path = 'tmp/pytorch_model.bin'
-ckpt = torch.load(model_path, map_location='cpu')
-diffusion_model.load_state_dict(ckpt, strict=False)
-
 # 下载文件
 os.makedirs('models', exist_ok=True)
 model_path = "models/relvid_mm_sd15_fbc_unet.pth"
@@ -170,6 +167,11 @@ model_path = "models/relvid_mm_sd15_fbc_unet.pth"
 if not os.path.exists(model_path):
     download_url_to_file(url='https://huggingface.co/aleafy/RelightVid/resolve/main/relvid_mm_sd15_fbc_unet.pth', dst=model_path)
 
+
+ckpt = torch.load(model_path, map_location='cpu')
+diffusion_model.load_state_dict(ckpt, strict=False)
+
+
 # import pdb; pdb.set_trace()
 
 # 更改全局临时目录