diff --git a/lmdeploy/utils.py b/lmdeploy/utils.py index de1bf04ef..206dd6d08 100644 --- a/lmdeploy/utils.py +++ b/lmdeploy/utils.py @@ -200,6 +200,7 @@ def get_model(pretrained_model_name_or_path: str, download_kwargs['token'] = token model_path = snapshot_download(pretrained_model_name_or_path, + ignore_patterns=['*.pth'], **download_kwargs) return model_path