diff --git a/bridge_former.py b/bridge_former.py index ba99614..0bdc7aa 100644 --- a/bridge_former.py +++ b/bridge_former.py @@ -84,7 +84,8 @@ class BridgeFormer(NNOperator): def _inference_from_text(self, text: List[str]): if self.model_name == "clip_initialized_model": - text_data = self.tokenizer(text, return_tensors='pt', padding='max_length', max_length=77).to(self.device) + text_data = self.tokenizer(text, return_tensors='pt', padding='max_length', max_length=77, + truncation=True).to(self.device) text_features = self.model.encode_text(text_data["input_ids"]) else: text_data = self.tokenizer(text, return_tensors='pt').to(self.device)