Spaces:
Paused
Paused
Update ldm/modules/encoders/modules.py
Browse files
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 |
-
|
| 178 |
-
|
| 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 |
|