Update download_model.py

download models with modelscope
2025-08-01 16:36:19 +08:00
parent 96d06b9b15
commit 948f1c2a06
1 changed files with 6 additions and 1 deletions
@@ -15,5 +15,10 @@ if __name__ == '__main__':
    if args.type == "huggingface":
        from huggingface_hub import snapshot_download
        snapshot_download(repo_id=args.name, local_dir=model_dir, local_dir_use_symlinks=False, resume_download=True)
+    elif args.type == "modelscope":
+        from modelscope import snapshot_download
+        snapshot_download(repo_id=args.name, local_dir=model_dir)
+    else:
+        raise ValueError(f"Invalid type: {args.type}")
    
-    print(f"model downloaded to {model_dir}")
+    print(f"model downloaded to {model_dir}")