From 581435ae3f6e159bfd9db754b9e85ce170595878 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 1 Feb 2024 03:23:07 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../transformers/modeling/modeling_auto.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/intel_extension_for_transformers/transformers/modeling/modeling_auto.py b/intel_extension_for_transformers/transformers/modeling/modeling_auto.py index 78669387d2ee..db02a130a82e 100644 --- a/intel_extension_for_transformers/transformers/modeling/modeling_auto.py +++ b/intel_extension_for_transformers/transformers/modeling/modeling_auto.py @@ -142,7 +142,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs): model_file = kwargs.get("model_file") gguf_model_file = hf_hub_download(pretrained_model_name_or_path, filename=model_file) - + if kwargs.get("model_type", False): model_type = kwargs.get("model_type") else: @@ -159,7 +159,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs): "dolly", "polyglot", "starcoder", "falcon", \ "bloom", "chatglm2", "chatglm", "baichuan", \ "mistral", "qwen", "phi", "whisper"] - + if model_type not in model_type_list: logger.error("Can't support this model_type. The supported model_type are: {}".format(model_type_list))