Browse Source
l
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
3 years ago
1 changed files with
1 additions and
1 deletions
-
bridge_former.py
|
|
@ -52,7 +52,7 @@ class BridgeFormer(NNOperator): |
|
|
weights_path=weights_path, |
|
|
weights_path=weights_path, |
|
|
model_name=self.model_name) |
|
|
model_name=self.model_name) |
|
|
self.transform_cfgs = configs(self.model_name) |
|
|
self.transform_cfgs = configs(self.model_name) |
|
|
|
|
|
|
|
|
|
|
|
self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False) |
|
|
self.model.eval() |
|
|
self.model.eval() |
|
|
|
|
|
|
|
|
def decoder_video(self, data: List[VideoFrame]): |
|
|
def decoder_video(self, data: List[VideoFrame]): |
|
|
|
