from @or-toledano: Reduce half of similarity muls after encoding (ope…

…nai#140)
rom1504 · Sep 26, 2022 · 61cd490 · 61cd490
1 parent e76d431
commit 61cd490
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clip/model.py b/clip/model.py
@@ -362,7 +362,7 @@ def forward(self, image, text):
         # cosine similarity as logits
         logit_scale = self.logit_scale.exp()
         logits_per_image = logit_scale * image_features @ text_features.t()
-        logits_per_text = logit_scale * text_features @ image_features.t()
+        logits_per_text = logits_per_image.t()
 
         # shape = [global_batch_size, global_batch_size]
         return logits_per_image, logits_per_text