28 lines
891 B
Markdown
28 lines
891 B
Markdown
|
# 天数智芯 天垓150 ASR（Transformers架构）
|
|||
|
|
|
|||
|
|
## 镜像构造
|
|||
|
|
```shell
|
|||
|
|
docker build -f ./Dockerfile.transformers-bi150 -t <your_image> .
|
|||
|
|
```
|
|||
|
|
其中，基础镜像 corex:4.3.8 通过联系天数智芯智铠100厂商技术支持可获取
|
|||
|
|
|
|||
|
|
## 使用说明
|
|||
|
|
|
|||
|
|
### 使用 FastAPI 启动ASR服务：
|
|||
|
|
例如：
|
|||
|
|
```shell
|
|||
|
|
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`目录下附带上了中文的测试音频和附带内容
|
|||
|
|
|
|||
|
|
```shell
|
|||
|
|
curl -X POST http://localhost:1111/transduce \
|
|||
|
|
-F "audio=@../sample_data/lei-jun-test.wav" \
|
|||
|
|
-F "lang=zh"
|
|||
|
|
```
|