Hugging Face's logo

Spaces:
SherryX
/
STAR
Running on Zero

App Files Community
8
xierui.0097 commited on
Commit
82b6fcd
·
1 Parent(s): e03120f

updates

Browse files
Files changed (1) hide show
  1. video_to_video/video_to_video_model.py +1 -0
video_to_video/video_to_video_model.py CHANGED
@@ -52,6 +52,7 @@ class VideoToVideo_sr():
52
  model_url = 'https://huggingface.co/SherryX/STAR/resolve/main/I2VGen-XL-based/heavy_deg.pt'
53
  download_model(model_url, cfg.model_path)
54
 
 
55
  load_dict = torch.load(cfg.model_path + '/heavy_deg.pt' , map_location='cpu')
56
  if 'state_dict' in load_dict:
57
  load_dict = load_dict['state_dict']
 
52
  model_url = 'https://huggingface.co/SherryX/STAR/resolve/main/I2VGen-XL-based/heavy_deg.pt'
53
  download_model(model_url, cfg.model_path)
54
 
55
+ # point to the weight
56
  load_dict = torch.load(cfg.model_path + '/heavy_deg.pt' , map_location='cpu')
57
  if 'state_dict' in load_dict:
58
  load_dict = load_dict['state_dict']