NguyenDinhHieu commited on
Commit
1feeb28
·
verified ·
1 Parent(s): b0cfd91

Update ldm/modules/encoders/modules.py

Browse files
Files changed (1) hide show
  1. ldm/modules/encoders/modules.py +2 -2
ldm/modules/encoders/modules.py CHANGED
@@ -174,8 +174,8 @@ class FrozenOpenCLIPEmbedder(AbstractEncoder):
174
 
175
  def forward(self, text):
176
  tokens = open_clip.tokenize(text, context_length=self.max_length).to(self.device)
177
- if tokens.dim() == 1:
178
- tokens = tokens.unsqueeze(0)
179
  z = self.encode_with_transformer(tokens)
180
  return z
181
 
 
174
 
175
  def forward(self, text):
176
  tokens = open_clip.tokenize(text, context_length=self.max_length).to(self.device)
177
+ if tokens.dim() == 1:
178
+ tokens = tokens.unsqueeze(0)
179
  z = self.encode_with_transformer(tokens)
180
  return z
181