From 158ad4484c5367c8c5c704985081bf5cf91e23a7 Mon Sep 17 00:00:00 2001 From: xujinling Date: Wed, 22 Jun 2022 18:51:08 +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 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]):