Spaces:

mehdidc
/

text_to_image_ddgan

Runtime error

Mehdi Cherti commited on Apr 17, 2023

Commit

2650632

1 Parent(s): deeb784

update

Files changed (1) hide show

clip_encoder.py CHANGED Viewed

@@ -23,8 +23,8 @@ class CLIPEncoder(nn.Module):
         self.model = model
         self.pretrained = pretrained
-        self.model, _, _ = open_clip.create_model_and_transforms(model)#, pretrained=pretrained)
-        print(self.model)
         self.output_size = self.model.transformer.width
     def forward(self, texts, return_only_pooled=False):

         self.model = model
         self.pretrained = pretrained
+        #self.model, _, _ = open_clip.create_model_and_transforms(model)#, pretrained=pretrained)
+        #print(self.model)
         self.output_size = self.model.transformer.width
     def forward(self, texts, return_only_pooled=False):