Fabrice-TIERCELIN commited on
Commit
7b5f999
·
verified ·
1 Parent(s): 452e82d
video_to_video/video_to_video_model.py CHANGED
@@ -50,8 +50,8 @@ class VideoToVideo_sr():
50
  # 确保 cfg.model_path 是文件夹路径,不要加上文件名
51
  cfg.model_path = opt.model_path
52
  # download weight
53
- #model_url = 'https://huggingface.co/SherryX/STAR/resolve/main/I2VGen-XL-based/heavy_deg.pt'
54
- #download_model(model_url, cfg.model_path)
55
 
56
  # 拼接完整路径
57
  model_file_path = os.path.join('pretrained_weight', 'I2VGen-XL-based', 'heavy_deg.pt')
 
50
  # 确保 cfg.model_path 是文件夹路径,不要加上文件名
51
  cfg.model_path = opt.model_path
52
  # download weight
53
+ model_url = 'https://huggingface.co/SherryX/STAR/resolve/main/I2VGen-XL-based/heavy_deg.pt'
54
+ download_model(model_url, cfg.model_path)
55
 
56
  # 拼接完整路径
57
  model_file_path = os.path.join('pretrained_weight', 'I2VGen-XL-based', 'heavy_deg.pt')