From 71c4b9e93a6a06274458443a2fa0d77829a231fb Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Thu, 9 Jun 2022 18:17:35 +0800 Subject: [PATCH] Debug Signed-off-by: Jael Gu --- auto_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto_transformers.py b/auto_transformers.py index 8f51d30..3d14f52 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -45,7 +45,7 @@ class AutoTransformers(NNOperator): self.model = AutoModel.from_pretrained(model_name) self.model.eval() except Exception as e: - model_list = get_model_list() + model_list = self.get_model_name() if model_name not in model_list: log.error(f"Invalid model name: {model_name}. Supported model names: {model_list}") else: