diff --git a/fastchat/model/model_adapter.py b/fastchat/model/model_adapter.py index 9625df6db..e773d5c80 100644 --- a/fastchat/model/model_adapter.py +++ b/fastchat/model/model_adapter.py @@ -360,9 +360,21 @@ def load_model( # lazy import so that modelscope is not required for normal use. try: from modelscope.hub.snapshot_download import snapshot_download + from modelscope.hub.errors import NotExistError if not os.path.exists(model_path): - model_path = snapshot_download(model_id=model_path, revision=revision) + try: + model_path = snapshot_download( + model_id=model_path, revision=revision + ) + except NotExistError as e: + # Default model revision could be "master" when use ModelScope. + if revision == "main": + revision = "master" + kwargs["revision"] = revision + model_path = snapshot_download( + model_id=model_path, revision=revision + ) except ImportError as e: warnings.warn( "Use model from www.modelscope.cn need pip install modelscope"