Merge pull request #9 from anyangml/fix/clip-ut

Fix: exp calc
anyangml · Jun 19, 2024 · a430dc8 · a430dc8
2 parents bba2c48 + caabfff
commit a430dc8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/ci_ut.yml b/.github/workflows/ci_ut.yml
@@ -32,5 +32,6 @@ jobs:
         uses: codecov/codecov-action@v4
         with:
           verbose: true
+          directory: clip
         env:
           CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
diff --git a/clip/clip/model.py b/clip/clip/model.py
@@ -30,6 +30,6 @@ def forward(self, text, image):
         embd_image = F.normalize(self.img_proj(encoded_image), dim=1)  # (B, D)
 
         # scaled pairwise cosine similarities (B, B)
-        logits = torch.mm(embd_text, embd_image.T) * torch.exp(self.temperature)
+        logits = torch.mm(embd_text, embd_image.T) * np.exp(self.temperature)
 
         return logits, logits.T  # text, image