diff --git a/bridge_former.py b/bridge_former.py index c25f1f5..025247b 100644 --- a/bridge_former.py +++ b/bridge_former.py @@ -52,7 +52,7 @@ class BridgeFormer(NNOperator): weights_path=weights_path, model_name=self.model_name) self.transform_cfgs = configs(self.model_name) - + self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False) self.model.eval() def decoder_video(self, data: List[VideoFrame]):