Spaces:
Configuration error
Configuration error
Update video_to_video/video_to_video_model.py (#4)
Browse files- Update video_to_video/video_to_video_model.py (19e6d48e7226eeb173587c46b4cc18e6d4daaf59)
Co-authored-by: Yinhong Liu <[email protected]>
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="
|
| 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')
|