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"
|
||
``` |