891 B
891 B
天数智芯 天垓150 ASR(Transformers架构)
镜像构造
docker build -f ./Dockerfile.transformers-bi150 -t <your_image> .
其中,基础镜像 corex:4.3.8 通过联系天数智芯智铠100厂商技术支持可获取
使用说明
使用 FastAPI 启动ASR服务:
例如:
docker run -dit -v /usr/src:/usr/src -v /lib/modules:/lib/modules --device=/dev/iluvatar0:/dev/iluvatar0 \
-v /mnt/contest_ceph/leaderboard/modelHubXC/openai-mirror/whisper-small:/model \
--network=host <your_image> \
main_transformers.py --model_dir /model --use_gpu --port 1111
具体参数代码设定可参考代码文件
测试ASR服务
项目根路径sample_data目录下附带上了中文的测试音频和附带内容
curl -X POST http://localhost:1111/transduce \
-F "audio=@../sample_data/lei-jun-test.wav" \
-F "lang=zh"