From 948f1c2a06f6da0c653c2edbc8c1843a3e6caa9c Mon Sep 17 00:00:00 2001 From: liferecords Date: Fri, 1 Aug 2025 16:36:19 +0800 Subject: [PATCH] Update download_model.py download models with modelscope --- tools/download_model.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tools/download_model.py b/tools/download_model.py index 03e36fd..32d7087 100755 --- a/tools/download_model.py +++ b/tools/download_model.py @@ -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}") \ No newline at end of file + print(f"model downloaded to {model_dir}")