From eac0c55d41d75b88052107355032b69ced8c80a7 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Thu, 12 Jan 2023 16:51:17 +0800 Subject: [PATCH] Remove test message Signed-off-by: Jael Gu --- auto_transformers.py | 1 - 1 file changed, 1 deletion(-) 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: