我
yangkang11111/GLM-OCR-Ascend-NPU
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

GLM-OCR-bf16 on vLLM-Ascend 0.18.0rc1

1. 简介

本文档记录 GLM-OCR-bf16 在 vLLM-Ascend 0.18.0rc1 环境的快速部署与验证结果。

GLM-OCR 是智谱 AI(Zhipu AI)发布的多模态 OCR 模型,基于 GLM-V encoder-decoder 架构,视觉编码器采用 CogViT,语言解码器为 GLM-0.5B,总参数量约 0.9B。该模型在 OmniDocBench V1.5 上取得 94.62 分(排名第一),支持文档、表格、公式、代码、印章等多种场景的文字识别。

相关获取地址:

  • 原始 HF 权重:https://huggingface.co/zai-org/GLM-OCR
  • ModelScope 权重:https://modelscope.cn/models/ZhipuAI/GLM-OCR
  • MLX 量化权重:https://huggingface.co/mlx-community/GLM-OCR-bf16
  • Docker Image(vLLM-Ascend 0.18.0rc1):quay.io/ascend/vllm-ascend:v0.18.0rc1

参考文档:

  • https://docs.vllm.ai/projects/ascend/zh-cn/v0.18.0/tutorials/models/Qwen3-Dense.html
  • https://docs.vllm.ai/projects/ascend/zh-cn/v0.18.0/user_guide/support_matrix/supported_models.html
  • https://arxiv.org/abs/2603.10910

2. 验证环境

组件版本
vllm-ascend0.18.0rc1
vllm0.18.0+empty
transformers4.57.6
torch-npu2.9.0.post1+gitee7ba04
torch2.9.0+cpu
  • NPU:2 逻辑卡(Ascend 910)
  • 模型路径:/opt/atomgit/weights/GLM-OCR-bf16
  • 服务端口:8000

3. 服务启动

启动前可先检查端口:

ss -lntp | grep ':8000 ' || true

已验证通过的启动命令:

export ASCEND_RT_VISIBLE_DEVICES=0,1
export VLLM_USE_MODELSCOPE=true
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export HCCL_BUFFSIZE=512
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1

vllm serve /opt/atomgit/weights/GLM-OCR-bf16 \
  --host 0.0.0.0 \
  --port 8000 \
  --data-parallel-size 1 \
  --tensor-parallel-size 1 \
  --seed 1024 \
  --served-model-name glm-ocr \
  --max-num-seqs 32 \
  --max-model-len 8192 \
  --max-num-batched-tokens 4096 \
  --trust-remote-code \
  --gpu-memory-utilization 0.85 \
  --no-enable-prefix-caching \
  --compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}'

关键参数说明:

  • --trust-remote-code:必须携带,GLM-OCR 使用自定义架构 GlmOcrForConditionalGeneration,需要 trust remote code
  • --tensor-parallel-size 1:0.9B 模型单卡即可运行
  • --max-model-len 8192:模型支持的最大上下文长度
  • --gpu-memory-utilization 0.85:HBM 利用率
  • --compilation-config FULL_DECODE_ONLY:ACL Graph 加速编译

4. Smoke 验证

基础检查:

curl -sf http://127.0.0.1:8000/v1/models
curl -sf http://127.0.0.1:8000/v1/chat/completions \
  -H "Content-Type: application/json" \
  -d '{
    "model": "glm-ocr",
    "messages": [
      {"role": "user", "content": "请用一句话介绍 GLM-OCR 模型。"}
    ],
    "temperature": 0,
    "max_tokens": 128
  }'

验证结果:

  • /v1/models 返回 200
  • /v1/chat/completions 返回 200
  • 文本生成正常,无乱码

5. 性能参考

测试条件:128 input / 128 output,API 级基准测试。

指标数值
Latency mean71.5 ms
Latency p99101.2 ms
Throughput25.14 req/s
Token throughput326.81 tok/s

注:测试并发度为 2,总请求数 10。实际性能受输入长度、并发度及编译参数影响,以上数据仅供参考。

压测命令:

# 需先确保 vLLM 服务已在 8000 端口运行
python scripts/bench_perf.py \
  --url http://localhost:8000/v1/chat/completions \
  --model glm-ocr \
  --max-tokens 128 \
  --latency-iters 5 \
  --throughput-requests 10 \
  --throughput-concurrency 2

6. 精度评测

6.1 验证方法

采用 NPU 与 CPU 基线对比的方式,对模型输出的 Logits 和 Hidden States 进行精度评估。验证指标包括:

  • max_abs_error: 最大绝对误差
  • mean_abs_error: 平均绝对误差
  • relative_error: 相对误差
  • cosine_similarity: 余弦相似度

6.2 验证结果

指标LogitsHidden States
max_abs_error0.1718750.250000
mean_abs_error0.0287620.018531
relative_error2.8886%7.4675%
cosine_similarity0.9999780.999889
threshold1.01.0
结果PASSPASS

结论:NPU 与 CPU 基线高度一致,cosine_similarity > 0.9998,验证通过。

说明:

  • 绝对误差(max_abs_error < 1.0)与余弦相似度(> 0.999)均满足通过标准。
  • 相对误差受 bfloat16 近零值影响偏大,仅作参考,不作为硬性判定依据。

运行精度评测:

python scripts/eval_accuracy.py \
  --model /opt/atomgit/weights/GLM-OCR-bf16 \
  --output accuracy_result.json

7. 推理脚本

项目提供 inference.py 一键推理脚本,支持本地 transformers 与 API 服务两种模式:

# 本地 transformers 模式(纯文本)
python inference.py --mode local \
  --model /opt/atomgit/weights/GLM-OCR-bf16 \
  --prompt "Text Recognition:"

# 本地 transformers 模式(图像 OCR)
python inference.py --mode local \
  --model /opt/atomgit/weights/GLM-OCR-bf16 \
  --image test.png \
  --prompt "Text Recognition:"

# API 服务模式(调用已部署的 vLLM 服务)
python inference.py --mode api \
  --url http://localhost:8000/v1/chat/completions \
  --model-name glm-ocr \
  --prompt "Text Recognition:"

8. 注意事项

  1. vLLM 原生支持

    • GlmOcrForConditionalGeneration 已在 vLLM 0.18.0 中注册,无需额外适配代码
    • 必须携带 --trust-remote-code,因为模型使用自定义架构类
  2. 多模态能力

    • GLM-OCR 是多模态 OCR 模型,支持图像输入
    • vLLM API 模式支持通过 base64 编码传入图像
    • 本地 transformers 模式支持直接传入 PIL Image 对象
  3. 显存占用

    • 0.9B 参数模型 BF16 精度下约占用 3-4 GB HBM
    • 单卡 64 GB Atlas 910 可轻松承载,无需张量并行
  4. 模型特点

    • GLM-OCR 使用 encoder-decoder 架构(非标准 causal LM)
    • 视觉编码器 CogViT + 语言解码器 GLM-0.5B
    • 支持多种 OCR 场景:文本、公式、表格、代码、印章

9. 参考信息

资源链接
原始 HF 权重https://huggingface.co/zai-org/GLM-OCR
ModelScope 权重https://modelscope.cn/models/ZhipuAI/GLM-OCR
技术报告https://arxiv.org/abs/2603.10910
vLLM-Ascend 文档https://docs.vllm.ai/projects/ascend/zh-cn/v0.18.0/
支持模型列表https://docs.vllm.ai/projects/ascend/zh-cn/v0.18.0/user_guide/support_matrix/supported_models.html