jinaai
/

jina-clip-implementation

🇪🇺 Region: EU

Model card Files Files and versions

bwang0911 commited on May 30, 2024

Commit

ff1a2e8

·

verified ·

1 Parent(s): 628d3ce

Update modeling_clip.py

Files changed (1) hide show

modeling_clip.py +1 -1

modeling_clip.py CHANGED Viewed

@@ -489,7 +489,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
                 processed_inputs.append(image)
-            processed_inputs = self.preprocess(processed_inputs).to(device)
             embeddings = self.get_image_features(processed_inputs)
             if normalize_embeddings:

                 processed_inputs.append(image)
+            processed_inputs = self.preprocess(processed_inputs)
             embeddings = self.get_image_features(processed_inputs)
             if normalize_embeddings: