blip_image_embeddings

radames commited on Apr 11, 2023

Commit

6639c4e

1 Parent(s): 562ec3c

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -52,4 +52,4 @@ class PreTrainedPipeline():
         with torch.no_grad():
             feature_vector = self.model(image, text, mode=parameters["mode"])[0,0].tolist()
         # postprocess the prediction
-        return {"feature_vector": feature_vector}

         with torch.no_grad():
             feature_vector = self.model(image, text, mode=parameters["mode"])[0,0].tolist()
         # postprocess the prediction
+        return feature_vector