Spaces:
Runtime error
Runtime error
Download
Browse files
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 |
-
|
| 54 |
-
|
| 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')
|