Commit 56479105 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent b97afd54
# MooER_pytorch
# MooER
## 论文
- https://arxiv.org/abs/2408.05101
......@@ -33,12 +33,12 @@ pip install -r requirements.txt
### Anaconda(方法三)
关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
```
DTK软件栈:dtk24,04,2
DTK软件栈:dtk24.04.2
Python:3.10
touch:2.3.0
torch:2.3.0
torchaudio:2.1.2
```
Tips:以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应
Tips:以上dtk驱动、python、pytorch等DCU相关工具版本需要严格一一对应
其它非深度学习库参照requirements.txt安装:
```
pip install -r requirements.txt
......@@ -48,7 +48,7 @@ pip install -r requirements.txt
## 训练
## 推理
1:首先从[ModelScope](https://modelscope.cn/models/MooreThreadsSpeech/MooER-MTL-5K)或者[HF-Mirror](https://hf-mirror.com/mtspeech/MooER-MTL-5K)下载官方提供的预训练模型
1:下载预训练模型MooER-MTL-5K,推荐使用scnet快速下载链接[MooER-MTL-5K](),官方下载地址[ModelScope](https://modelscope.cn/models/MooreThreadsSpeech/MooER-MTL-5K)或者[HF-Mirror](https://hf-mirror.com/mtspeech/MooER-MTL-5K)。
```
# 使用ModelScope
git lfs clone https://modelscope.cn/models/MooreThreadsSpeech/MooER-MTL-5K
......@@ -57,19 +57,10 @@ git lfs clone https://modelscope.cn/models/MooreThreadsSpeech/MooER-MTL-5K
git lfs clone https://hf-mirror.com/mtspeech/MooER-MTL-5K
```
将下载后的文件放置在 `pretrained_models` 文件夹中。
```shell
cp MooER-MTL-5K/* pretrained_models
```
2:下载 [`Qwen2-7B-Instruct`](https://modelscope.cn/models/qwen/qwen2-7b-instruct)
```
# 使用ModelScope
git lfs clone https://modelscope.cn/models/qwen/qwen2-7b-instruct
# 使用HuggingFace
git lfs clone https://hf-mirror.com/Qwen/Qwen2-7B-Instruct
```
2:下载Qwen2-7B-Instruct,推荐使用scnet快速下载链接 [`Qwen2-7B-Instruct`](http://113.200.138.88:18080/aimodels/Qwen2-7B-Instruct)官方下载地址[ModelScope](https://modelscope.cn/models/qwen/qwen2-7b-instruct)或者[HF-Mirror](https://hf-mirror.com/Qwen/Qwen2-7B-Instruct)
将下载后的文件放在 `pretrained_models/Qwen2-7B-Instruct` 文件夹中。
......@@ -126,15 +117,13 @@ export PYTHONIOENCODING=UTF-8
export LC_ALL=C
export PYTHONPATH=$PWD/src:$PYTHONPATH
```
- 同时进行ASR和AST:
**同时进行ASR和AST:**
```
# 使用指定的音频文件
python inference.py --wav_path /path/to/your_audio_file
```
上述命令会调用一个多任务Mooer大模型,同时输出语音识别和语音翻译的结果。如果运行成功,将在终端看到如下结果。<br>
![asr-adn-ast](images/asr-and-ast.png)
- 指定语音识别模型,仅输出识别结果:
<br>
**指定语音识别模型,仅输出识别结果:**
```
python inference.py --task asr \
--cmvn_path pretrained_models/paraformer_encoder/am.mvn \
......@@ -144,10 +133,8 @@ python inference.py --task asr \
--lora_dir pretrained_models/asr/lora_weights \
--wav_path /path/to/your_audio_file
```
上述命令会仅输出语音识别的结果。如果运行成功,将在终端看到如下结果。<br>
![only-asr](images/only-asr.png)
- 指定语音翻译模型,仅输出中译英结果
<br>
**指定语音翻译模型,仅输出中译英结果**
```
python inference.py --task ast \
--cmvn_path pretrained_models/paraformer_encoder/am.mvn \
......@@ -157,15 +144,28 @@ python inference.py --task ast \
--lora_dir pretrained_models/ast/lora_weights \
--wav_path /path/to/your_audio_file
```
上述命令会仅输出语音翻译的结果。如果运行成功,将在终端看到如下结果。<br>
![only-ast](images/only-ast.png)
## result
**ASR和AST**
```
ASR: 欢迎使用由摩尔线程开发的基于大语言模型的语音识别及语音翻译系统
AST: Welcome to use the voice recognition and voice translation system based on the large language model developed by Moore Threads.
```
**ASR**
```
ASR: 欢迎使用由摩尔线程开发的基于大语言模型的语音识别及语音翻译系统
```
**AST**
```
AST: Welcome to use the voice recognition and voice translation system based on the large language model developed by Moore Threads.
```
### 精度
## 应用场景
### 算法分类
语音识别、语音翻译
`语音识别,语音翻译`
### 热点应用行业
语音识别、语音翻译、教育医疗
`教育,医疗,科研`
## 源码仓库及问题反馈
https://developer.sourcefind.cn/codes/modelzoo/mooer_pytorch
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment