From d038cfaf5927dc8b621c8d6f6d61df2955ec7638 Mon Sep 17 00:00:00 2001
From: Canberk Baykal <38884482+johnberg1@users.noreply.github.com>
Date: Tue, 29 Jun 2021 16:13:53 +0300
Subject: [PATCH] Update model.py

---
 clip/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clip/model.py b/clip/model.py
index 422a34a..7c7832d 100644
--- a/clip/model.py
+++ b/clip/model.py
@@ -365,7 +365,7 @@ class CLIP(nn.Module):
         logits_per_text = logit_scale * text_features @ image_features.t()
 
         # shape = [global_batch_size, global_batch_size]
-        return logits_per_image, logits_per_text
+        return logits_per_image, logits_per_text, logit_scale
 
 
 def convert_weights(model: nn.Module):