diff --git a/clip/model.py b/clip/model.py index 232b7792e..8268402c1 100644 --- a/clip/model.py +++ b/clip/model.py @@ -365,6 +365,7 @@ def forward(self, image, text): # cosine similarity as logits logit_scale = self.logit_scale.exp() + logit_scale = torch.clamp(logit_scale, max=100) logits_per_image = logit_scale * image_features @ text_features.t() logits_per_text = logits_per_image.t()