引言

本案例给出微软语音模型Beats在NPU环境部署，并基于torch_npu执行推理任务的迁移实践。

使用约束

硬件设备

一、环境准备

安装依赖包： pip install -r requirements.txt

二、下载官方代码和权重

2.1 下载官方代码

git clone https://github.com/microsoft/unilm.git

2.2 下载官方权重

https://github.com/microsoft/unilm/tree/master/beats 官方代码仓开源了不同版本模型的权重文件，选择适配自己任务的模型下载

三、运行指导

3.1 修改推理脚本和项目代码

替换官方代码UniLM/beats/BEATs.py为本代码仓中的同名文件。

拷贝infer_npu.py到UniLM目录下。

3.2 执行推理

python infer_npu.py