add different hubs usage
This commit is contained in:
@ -81,6 +81,8 @@ def main():
|
|||||||
trust_remote_code=True,
|
trust_remote_code=True,
|
||||||
remote_code="./model.py",
|
remote_code="./model.py",
|
||||||
device="cuda:0",
|
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"
|
wav_path = f"{model.model_path}/example/zh.mp3"
|
||||||
|
|||||||
@ -81,6 +81,7 @@ def main():
|
|||||||
trust_remote_code=True,
|
trust_remote_code=True,
|
||||||
remote_code="./model.py",
|
remote_code="./model.py",
|
||||||
device="cuda:0",
|
device="cuda:0",
|
||||||
|
hub="ms"
|
||||||
)
|
)
|
||||||
|
|
||||||
wav_path = f"{model.model_path}/example/zh.mp3"
|
wav_path = f"{model.model_path}/example/zh.mp3"
|
||||||
|
|||||||
2
demo1.py
2
demo1.py
@ -18,6 +18,8 @@ def main():
|
|||||||
trust_remote_code=True,
|
trust_remote_code=True,
|
||||||
remote_code="./model.py",
|
remote_code="./model.py",
|
||||||
device=device,
|
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"
|
wav_path = f"{model.model_path}/example/zh.mp3"
|
||||||
|
|||||||
Reference in New Issue
Block a user