diff --git a/auto_transformers.py b/auto_transformers.py index ea88ad5..bf278d7 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -73,7 +73,6 @@ class AutoTransformers(NNOperator): else: self.device = 'cuda' if torch.cuda.is_available() else 'cpu' self.model_name = self.map_model_names(model_name) - print(1111, self.model_name) if tokenizer: self.tokenizer = tokenizer else: