l

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · ddfb0178a8
1 changed files with 1 additions and 1 deletions
--- a/bridge_former.py
+++ b/bridge_former.py
@ -84,7 +84,7 @@ class BridgeFormer(NNOperator):

        if self.model_name == "clip_initialized_model":
            text_data = tokenize(text).to(self.device)
-            text_features = self.model.encode_text(text_data["input_ids"])
+            text_features = self.model.encode_text(text_data)
        else:
            text_data = self.tokenizer(text, return_tensors='pt').to(self.device)
            text_features = self.model.compute_text(text_data)