logo
Browse Source

Remove test message

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
eac0c55d41
  1. 1
      auto_transformers.py

1
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:

Loading…
Cancel
Save