Mehdi Cherti commited on
Commit
df793a8
·
1 Parent(s): 2650632
Files changed (1) hide show
  1. 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 = self.model.transformer.width
 
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