updates

video_to_video/video_to_video_model.py

@@ -47,13 +47,19 @@ class VideoToVideo_sr():
         generator = generator.to(self.device)
         generator.eval()
 
+        # 确保 cfg.model_path 是文件夹路径，不要加上文件名
         cfg.model_path = opt.model_path
         # download weight
         model_url = 'https://huggingface.co/SherryX/STAR/resolve/main/I2VGen-XL-based/heavy_deg.pt'
         download_model(model_url, cfg.model_path)
 
-        # point to the weight
-        load_dict = torch.load(cfg.model_path + '/heavy_deg.pt' , map_location='cpu')
+        # 拼接完整路径
+        model_file_path = os.path.join(cfg.model_path, 'heavy_deg.pt')
+        print('model_file_path:', model_file_path)
+
+        # 加载模型
+        load_dict = torch.load(model_file_path, map_location='cpu')
+
         if 'state_dict' in load_dict:
             load_dict = load_dict['state_dict']
         ret = generator.load_state_dict(load_dict, strict=False)