Spaces:

mehdidc
/

text_to_image_ddgan

Runtime error

Mehdi Cherti commited on Apr 17, 2023

Commit

857bdd2

1 Parent(s): c84aae2

update

Files changed (1) hide show

clip_encoder.py CHANGED Viewed

@@ -23,7 +23,7 @@ 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

         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