Skip to content

Commit 1125bad

Browse files
committed
fix
1 parent e06b850 commit 1125bad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llm2clip/eva_clip/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def set_grad_checkpointing(self, enable=True):
245245
def forward(self, text, return_all_features: bool=False, l2_norm: bool=True):
246246
if l2_norm:
247247
x = torch.nn.functional.normalize(text, p=2, dim=-1)
248-
x = self.text_adaptor(text)
248+
x = self.text_adaptor(x)
249249
return x
250250

251251

0 commit comments

Comments
 (0)