fix

Author: raytrun

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