Debug

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 71c4b9e93a
1 changed files with 1 additions and 1 deletions
--- 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: