Mirrors/dots.ocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update download_model.py

Browse Source 
download models with modelscope
This commit is contained in:
liferecords
2025-08-01 16:36:19 +08:00
committed by GitHub
parent 96d06b9b15
commit 948f1c2a06
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/download_model.py
+5
View File
@@ -15,5 +15,10 @@ if __name__ == '__main__':
if args.type == "huggingface": if args.type == "huggingface":
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
snapshot_download(repo_id=args.name, local_dir=model_dir, local_dir_use_symlinks=False, resume_download=True) 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}")