冬
gcw_IDzXRVNw/opus-mt_tiny_tur-eng-ascend
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

opus-mt_tiny_tur-eng Ascend NPU 部署指南

项目简介

opus-mt_tiny_tur-eng 是 Helsinki-NLP 系列的多语言翻译模型，专门用于土耳其语(TUR)到英语(ENG)的翻译任务。该模型基于 Transformer 架构，参数量约 55M。

特性

  • 支持 Ascend NPU 推理加速
  • CPU 与 NPU 精度对比测试（输出完全一致）
  • 土耳其语到英语翻译
  • 兼容 HuggingFace transformers
  • 支持批量翻译

环境要求

  • 硬件：华为 Ascend 910 系列 NPU
  • CANN：8.0.RC1 或更高版本
  • PyTorch：2.0+ 并带有 torch_npu
  • Docker：容器名称 test-modelagent
  • transformers：4.8+

目录结构

opus-mt_tiny_tur-eng-ascend/
├── inference.py          # 推理测试脚本
├── log.txt               # 测试日志
├── README.md             # 本文档
├── test_sample.txt       # 测试样例
├── inference_result.json # 推理结果
└── precision_result.json # 精度测试结果

部署步骤

1. 进入容器

docker exec -it test-modelagent bash

2. 设置环境变量

source /usr/local/Ascend/ascend-toolkit/set_env.sh

3. 准备模型文件

模型文件位于 /data/ysws/agentsp/5-17/opus-mt_tiny_tur-eng/ 目录下：

  • model.safetensors - 模型权重
  • config.json - 模型配置
  • tokenizer.json / vocab.json - 分词器文件
  • source.spm / target.spm - SentencePiece 模型

4. 安装依赖

pip install transformers torch_npu -i https://huaweimirror.com.cn/simple

Usage

Method 1: Normal Inference Mode

Run the inference script for translation:

cd /data/ysws/agentsp/5-17/opus-mt_tiny_tur-eng-ascend/

python3 inference.py

方式二：精度测试模式 (CPU vs NPU)

运行精度对比测试，验证 NPU 计算结果与 CPU 一致性：

cd /data/ysws/agentsp/5-17/opus-mt_tiny_tur-eng-ascend/

python3 inference.py all

测试验证

精度测试结果

指标实测值阈值状态
翻译一致性100%100%PASS
输出匹配TrueTruePASS

性能数据

操作耗时
CPU 推理时间0.153s
NPU 推理时间0.065s
加速比2.36x

推理结果示例

输入 (土耳其语)输出 (英语)
"Merhaba, bugün nasılsınız?""Hello, how are you doing today?"
"Seni görmek çok güzel."英语翻译
"Otomatik çeviri çok faydalı."英语翻译

结果: CPU 和 NPU 输出的翻译结果完全一致

测试日志

============================================================
OPUS-MT-TINY-TUR-ENG NPU Test
Model: Helsinki-NLP/opus-mt_tiny_tur-eng
Output: /data/ysws/agentsp/5-17/opus-mt_tiny_tur-eng-ascend
============================================================

============================================================
OPUS-MT-TINY-TUR-ENG Inference Test (NPU)
============================================================
Device: npu:0
Model: /data/ysws/agentsp/5-17/opus-mt_tiny_tur-eng/Helsinki-NLP/opus-mt_tiny_tur-eng

Loading tokenizer...
Loading model...
Loading weights: 100%|██████████| 151/151 [00:00<00:00, 4813.13it/s]

Input text: ['Merhaba, bugün nasılsınız?']
Input shape: torch.Size([1, 7])
Generated text: ['Hello, how are you doing today?']
Inference time: 0.964s

============================================================
Precision Test (CPU vs NPU)
============================================================

Using device: npu:0
Loading tokenizer...
Loading model on CPU...
Loading weights: 100%|██████████| 151/151 [00:00<00:00, 4349.62it/s]
Running inference on CPU...

Loading model on npu:0...
Loading weights: 100%|██████████| 151/151 [00:00<00:00, 4284.16it/s]
Running inference on NPU...

CPU inference time: 0.153s
NPU inference time: 0.065s
Speedup: 2.36x
CPU output: ['Hello, how are you doing today?']
NPU output: ['Hello, how are you doing today?']
Output texts match: True
Status: PASS

Python API 使用示例

基本翻译

import torch
from transformers import MarianMTModel, MarianTokenizer

MODEL_DIR = "/data/ysws/agentsp/5-17/opus-mt_tiny_tur-eng/Helsinki-NLP/opus-mt_tiny_tur-eng"

tokenizer = MarianTokenizer.from_pretrained(MODEL_DIR)
model = MarianMTModel.from_pretrained(MODEL_DIR)
model = model.to("npu:0")
model.eval()

src_texts = ["Merhaba, bugün nasılsınız?"]
inputs = tokenizer(src_texts, return_tensors="pt", padding=True)
inputs = {k: v.to("npu:0") for k, v in inputs.items()}

with torch.no_grad():
    outputs = model.generate(**inputs)

translations = tokenizer.batch_decode(outputs, skip_special_tokens=True)
print(translations)  # ['Hello, how are you doing today?']

批量翻译

src_texts = [
    "Merhaba, bugün nasılsınız?",
    "Seni görmek çok güzel.",
    "Otomatik çeviri çok faydalı."
]

inputs = tokenizer(src_texts, return_tensors="pt", padding=True)
inputs = {k: v.to("npu:0") for k, v in inputs.items()}

with torch.no_grad():
    outputs = model.generate(**inputs)

translations = tokenizer.batch_decode(outputs, skip_special_tokens=True)
for src, tgt in zip(src_texts, translations):
    print(f"{src} -> {tgt}")

模型结构

  • 架构类型: MarianMT（Transformer）
  • 编码器: 6 层 Transformer
  • 解码器: 6 层 Transformer
  • 隐藏层维度: 512
  • 注意力头数: 8
  • 参数量: ~55M
  • 源语言: 土耳其语（tur）
  • 目标语言: 英语（eng）

推理参数配置

从 config.json 提取的关键参数:

{
  "hidden_size": 512,
  "encoder_layers": 6,
  "decoder_layers": 6,
  "encoder_attention_heads": 8,
  "decoder_attention_heads": 8,
  "d_model": 512,
  "vocab_size": 58101
}

常见问题

Q: 精度测试失败?

A: 检查 NPU 驱动是否正确安装，确保 CANN 环境变量已 source。

Q: 如何提高推理速度?

A: 使用批处理可以显著提高吞吐量。另外，首次推理会有编译开销，后续推理会更快。

Q: 模型支持哪些语言方向?

A: 本模型专门用于土耳其语到英语的翻译。如需其他语言对，请访问 Helsinki-NLP 模型库。

参考链接

  • 原始模型: https://huggingface.co/Helsinki-NLP/opus-mt_tiny_tur-eng
  • MarianMT: https://huggingface.co/transformers/model_doc/marian.html
  • HuggingFace Transformers: https://huggingface.co/transformers

许可证

本项目遵循 CC-BY-NC 4.0 许可证