Hugging Face's logo

Spaces:
SherryX
/
STAR
Configuration error

App Files Community
11

Update video_to_video/video_to_video_model.py

#4
by Solitude-liu - opened
base: refs/heads/main
←
from: refs/pr/4
Discussion Files changed
+1
-1
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')