Browse Source
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
auto_transformers.py
|
|
@ -45,7 +45,7 @@ class AutoTransformers(NNOperator): |
|
|
self.model = AutoModel.from_pretrained(model_name) |
|
|
self.model = AutoModel.from_pretrained(model_name) |
|
|
self.model.eval() |
|
|
self.model.eval() |
|
|
except Exception as e: |
|
|
except Exception as e: |
|
|
model_list = get_model_list() |
|
|
|
|
|
|
|
|
model_list = self.get_model_name() |
|
|
if model_name not in model_list: |
|
|
if model_name not in model_list: |
|
|
log.error(f"Invalid model name: {model_name}. Supported model names: {model_list}") |
|
|
log.error(f"Invalid model name: {model_name}. Supported model names: {model_list}") |
|
|
else: |
|
|
else: |
|
|
|
