diff --git a/README.md b/README.md index fb14a88..2753219 100755 --- a/README.md +++ b/README.md @@ -1015,6 +1015,9 @@ pip install -e . > 💡**Note:** Please use a directory name without periods (e.g., `DotsOCR` instead of `dots.ocr`) for the model save path. This is a temporary workaround pending our integration with Transformers. ```shell python3 tools/download_model.py + +# with modelscope +python3 tools/download_model.py --type modelscope ``` diff --git a/requirements.txt b/requirements.txt index d7dedd4..a327bb4 100755 --- a/requirements.txt +++ b/requirements.txt @@ -7,5 +7,6 @@ openai qwen_vl_utils transformers==4.51.3 huggingface_hub +modelscope flash-attn==2.8.0.post2 -accelerate \ No newline at end of file +accelerate 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}")