本文档记录 dengcao/gte-small-zh 在昇腾 NPU（Ascend910）环境的快速部署与验证结果。
BertModel 文本向量模型，基于 HuggingFace transformers 框架，支持文本特征提取与语义匹配。
相关获取地址：
参考文档：
| 组件 | 版本 |
|---|---|
torch | 2.5.1 |
torch_npu | 2.5.1 |
transformers | >=4.48.0 |
CANN | 8.5.RC1 |
512512PyTorch + transformerspip install transformers torch torchvision sentence-transformersimport torch
from transformers import AutoModel, AutoTokenizer
device = torch.device("npu:0" if torch.npu.is_available() else "cpu")
tokenizer = AutoTokenizer.from_pretrained("dengcao/gte-small-zh")
model = AutoModel.from_pretrained("dengcao/gte-small-zh")
model = model.to(device).eval()
sentences = ["今天天气怎么样", "如何学习编程"]
inputs = tokenizer(sentences, padding=True, truncation=True, return_tensors="pt")
inputs = {k: v.to(device) for k, v in inputs.items()}
with torch.no_grad():
outputs = model(**inputs)
embeddings = outputs.last_hidden_state.mean(dim=1)
print(f"Embedding shape: {embeddings.shape}")python3 inference.py验证结果：
npu:0测试条件：FP32 / batch=8 / warmup=5 / timed=100 runs，Ascend910 单卡。
| 指标 | 数值 |
|---|---|
| 平均推理时间 | 3.72 ms |
| QPS（每秒查询） | 2149.11 |
| 测试次数 | 100 |
NPU 与 CPU 输出对比，使用 8 条测试文本，比较 embedding 余弦相似度。
| 指标 | 数值 |
|---|---|
| 平均余弦相似度 | 0.999999 |
| 最低余弦相似度 | 0.999999 |
| 最大向量差异 | 0.002519 |
| 平均向量差异 | 0.000557 |
| 结论 | PASS |
pooler_output