Hugging Face's logo

Spaces:
SherryX
/
STAR
Running on Zero

App Files Community
5
Solitude-liu commited on
Commit
19e6d48
·
verified ·
1 Parent(s): 46518d3

Update video_to_video/video_to_video_model.py

Browse files
Files changed (1) hide show
  1. video_to_video/video_to_video_model.py +1 -1
video_to_video/video_to_video_model.py CHANGED
@@ -37,7 +37,7 @@ class VideoToVideo_sr():
37
  self.device = device # torch.device(f'cuda:0')
38
 
39
  # text_encoder
40
- text_encoder = FrozenOpenCLIPEmbedder(device=self.device, pretrained="/home/test/Workspace/yhliu/VSR/ours/checkpoints/open_clip_pytorch_model.bin")
41
  text_encoder.model.to(self.device)
42
  self.text_encoder = text_encoder
43
  logger.info(f'Build encoder with FrozenOpenCLIPEmbedder')
 
37
  self.device = device # torch.device(f'cuda:0')
38
 
39
  # text_encoder
40
+ text_encoder = FrozenOpenCLIPEmbedder(device=self.device, pretrained="laion2b_s32b_b79k")
41
  text_encoder.model.to(self.device)
42
  self.text_encoder = text_encoder
43
  logger.info(f'Build encoder with FrozenOpenCLIPEmbedder')