l

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · 158ad4484c
1 changed files with 1 additions and 1 deletions
--- 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]):