flavour
/

clippy-dinov2-small-bge-small-en

Feature Extraction

Model card Files Files and versions Community

marcusinthesky commited on Sep 4, 2023

Commit

dcbff95

·

1 Parent(s): e3f253b

Upload model

Files changed (1) hide show

modelling.py +1 -1

modelling.py CHANGED Viewed

@@ -157,7 +157,7 @@ class VTDEModel(VisionTextDualEncoderModel):
             loss = clip_loss(logits_per_text)
         if not return_dict:
-            output = (logits_per_image, logits_per_text, text_embeds, image_embeds, text_outputs, vision_outputs)
             return ((loss,) + output) if loss is not None else output
         return CLIPOutput(

             loss = clip_loss(logits_per_text)
         if not return_dict:
+            output = (logits_per_image, logits_per_text, text_embeds, image_embeds, text_embeds, image_embeds)
             return ((loss,) + output) if loss is not None else output
         return CLIPOutput(