updatas

video_to_video/video_to_video_model.py

@@ -52,7 +52,7 @@ class VideoToVideo_sr():
         model_url = 'https://huggingface.co/SherryX/STAR/resolve/main/I2VGen-XL-based/heavy_deg.pt'
         download_model(model_url, cfg.model_path)
 
-        load_dict = torch.load(cfg.model_path, map_location='cpu')
+        load_dict = torch.load(cfg.model_path + '/heavy_deg.pt' , map_location='cpu')
         if 'state_dict' in load_dict:
             load_dict = load_dict['state_dict']
         ret = generator.load_state_dict(load_dict, strict=False)