From ddfb0178a86e794be56171f1a4424ef6dd055f36 Mon Sep 17 00:00:00 2001 From: xujinling Date: Wed, 22 Jun 2022 18:26:41 +0800 Subject: [PATCH] l Signed-off-by: xujinling --- bridge_former.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bridge_former.py b/bridge_former.py index 2109cb3..c25f1f5 100644 --- 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)