diff --git a/README.md b/README.md index a623f79..4cc6715 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,6 @@ # enginex-hygon-asr -运行于【海光 DCU】系列算力卡的【语音识别】引擎,基于 transformer 架构,支持 SenseVoice、Whisper 等最新流行模型 \ No newline at end of file +运行于【海光 DCU】系列算力卡的【语音识别】引擎,基于 transformer 架构,支持 SenseVoice、Whisper 等最新流行模型 + +支持模型列表: +- microsoft/speecht5_asr \ No newline at end of file diff --git a/enginex/speecht5_asr.md b/enginex/speecht5_asr.md new file mode 100644 index 0000000..a9f825b --- /dev/null +++ b/enginex/speecht5_asr.md @@ -0,0 +1,10 @@ +# 运行方式 + +```python +docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-centos7.6-dtk24.04-py310 + +docker run -it --network=host --ipc=host --name=your_container_name --shm-size=32G --device=/dev/kfd --device=/dev/mkfd --device=/dev/dri -v /opt/hyhal:/opt/hyhal:ro -v /path/your_code_data/:/path/your_code_data/ --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310 /bin/bash + +cd /path/your_code_data/ +pip3 install -r requirements.txt +```