Spaces:
Runtime error
Runtime error
Mehdi Cherti
commited on
Commit
·
df793a8
1
Parent(s):
2650632
update
Browse files- clip_encoder.py +2 -1
clip_encoder.py
CHANGED
@@ -25,7 +25,8 @@ class CLIPEncoder(nn.Module):
|
|
25 |
self.pretrained = pretrained
|
26 |
#self.model, _, _ = open_clip.create_model_and_transforms(model)#, pretrained=pretrained)
|
27 |
#print(self.model)
|
28 |
-
self.output_size =
|
|
|
29 |
|
30 |
def forward(self, texts, return_only_pooled=False):
|
31 |
device = next(self.parameters()).device
|
|
|
25 |
self.pretrained = pretrained
|
26 |
#self.model, _, _ = open_clip.create_model_and_transforms(model)#, pretrained=pretrained)
|
27 |
#print(self.model)
|
28 |
+
self.output_size = 1024
|
29 |
+
#self.output_size = self.model.transformer.width
|
30 |
|
31 |
def forward(self, texts, return_only_pooled=False):
|
32 |
device = next(self.parameters()).device
|