diff --git a/README.md b/README.md index 4ed2730..58e695a 100644 --- a/README.md +++ b/README.md @@ -81,6 +81,8 @@ def main(): trust_remote_code=True, remote_code="./model.py", device="cuda:0", + # hub:download models from ms (for ModelScope) or hf (for Hugging Face). + hub="hf" ) wav_path = f"{model.model_path}/example/zh.mp3" diff --git a/README_zh.md b/README_zh.md index a83ff49..75dc620 100644 --- a/README_zh.md +++ b/README_zh.md @@ -81,6 +81,7 @@ def main(): trust_remote_code=True, remote_code="./model.py", device="cuda:0", + hub="ms" ) wav_path = f"{model.model_path}/example/zh.mp3" diff --git a/demo1.py b/demo1.py index 19cc270..f91a07b 100644 --- a/demo1.py +++ b/demo1.py @@ -18,6 +18,8 @@ def main(): trust_remote_code=True, remote_code="./model.py", device=device, + # hub:download models from ms (for ModelScope) or hf (for Hugging Face). + hub="ms" ) wav_path = f"{model.model_path}/example/zh.mp3"