ignore *.pth when download model from model hub (#2426)

* ignore *.pth when download model from model hub * update
InternLM · Sep 5, 2024 · 4866410 · 4866410
1 parent aa44e3d
commit 4866410
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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