Skip to content
This repository has been archived by the owner on Oct 25, 2024. It is now read-only.

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Feb 1, 2024
1 parent 7a112e3 commit 581435a
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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))

Expand Down

0 comments on commit 581435a

Please sign in to comment.