fix

microsoft · Nov 25, 2024 · ca5634f · ca5634f
1 parent 1125bad
commit ca5634f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llm2clip/eva_clip/model.py b/llm2clip/eva_clip/model.py
@@ -244,8 +244,8 @@ def set_grad_checkpointing(self, enable=True):
 
     def forward(self, text, return_all_features: bool=False, l2_norm: bool=True):
         if l2_norm:
-            x = torch.nn.functional.normalize(text, p=2, dim=-1)
-        x = self.text_adaptor(x)
+            text = torch.nn.functional.normalize(text, p=2, dim=-1)
+        x = self.text_adaptor(text)
         return x