Browse Source
Remove test message
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
0 additions and
1 deletions
-
auto_transformers.py
|
@ -73,7 +73,6 @@ class AutoTransformers(NNOperator): |
|
|
else: |
|
|
else: |
|
|
self.device = 'cuda' if torch.cuda.is_available() else 'cpu' |
|
|
self.device = 'cuda' if torch.cuda.is_available() else 'cpu' |
|
|
self.model_name = self.map_model_names(model_name) |
|
|
self.model_name = self.map_model_names(model_name) |
|
|
print(1111, self.model_name) |
|
|
|
|
|
if tokenizer: |
|
|
if tokenizer: |
|
|
self.tokenizer = tokenizer |
|
|
self.tokenizer = tokenizer |
|
|
else: |
|
|
else: |
|
|