| 项目 | 内容 |
|---|---|
| 模型名称 | all-mpnet-base-v2 |
| 原始模型 | https://huggingface.co/sentence-transformers/all-mpnet-base-v2 |
| 任务类型 | text embedding |
| 硬件环境 | Ascend 910 NPU |
| 框架 | PyTorch 2.x + torch_npu + sentence-transformers |
本项目将 all-mpnet-base-v2 模型适配至昇腾 NPU 平台，通过 ModelScope 下载权重，torch_npu 实现 CPU/NPU 精度对齐。
模型权重通过 ModelScope 官方渠道下载，不提交至本仓库。
本仓库仅包含推理代码与适配脚本，不提交任何模型权重文件（.bin, .safetensors, .pt, .ckpt）。
pip install modelscope
python inference.py| 指标 | 值 |
|---|---|
| Cosine Similarity | {cos} |
| Mean Abs Diff | {mad} |
| Match | {match} |
| 环境 | 延迟 |
|---|---|
| CPU | {cpu_ms}ms |
| NPU | {npu_ms}ms |
模型在昇腾 NPU 上推理结果与 CPU 一致，精度满足要求。