跳转至

English

PaddleOCR-VL-0.9B

一、环境准备

1.1 支持情况

推荐硬件配置: - 显存:8GB显存及以上 - 共享内存:4G及以上

1.2 安装fastdeploy

安装流程参考文档 FastDeploy GPU 安装

二、如何使用

2.1 基础:启动服务

示例1: 3060上单卡部署16K上下文的服务

python -m fastdeploy.entrypoints.openai.api_server \
    --model PaddlePaddle/PaddleOCR-VL \
    --port 8185 \
    --metrics-port 8186 \
    --engine-worker-queue-port 8187 \
    --max-model-len 16384 \
    --max-num-batched-tokens 16384 \
    --gpu-memory-utilization 0.8 \
    --max-num-seqs 128

示例2: 4090上单卡部署16K上下文的服务

python -m fastdeploy.entrypoints.openai.api_server \
    --model PaddlePaddle/PaddleOCR-VL \
    --port 8185 \
    --metrics-port 8186 \
    --engine-worker-queue-port 8187 \
    --max-model-len 16384 \
    --max-num-batched-tokens 16384 \
    --gpu-memory-utilization 0.7 \
    --max-num-seqs 256

示例3: A100上单卡部署16K上下文的服务

python -m fastdeploy.entrypoints.openai.api_server \
    --model PaddlePaddle/PaddleOCR-VL \
    --port 8185 \
    --metrics-port 8186 \
    --engine-worker-queue-port 8187 \
    --max-model-len 16384 \
    --max-num-batched-tokens 16384 \
    --gpu-memory-utilization 0.7 \
    --max-num-seqs 256

示例是可以稳定运行的一组配置,同时也能得到比较好的性能。 如果对精度、性能有进一步的要求,请继续阅读下面的内容。

2.2 进阶:如何获取更优性能

2.2.1 评估应用场景,正确设置参数

上下文长度 - 参数: --max-model-len - 描述: 控制模型可处理的最大上下文长度。 - 推荐: 更长的上下文会导致吞吐降低,根据实际情况设置,PaddleOCR-VL-0.9B最长支持16k(16,384)长度的上下文。

⚠️ 注:更长的上下文会显著增加GPU显存需求,设置更长的上下文之前确保硬件资源是满足的。

最大序列数量 - 参数: --max-num-seqs - 描述: 控制服务可以处理的最大序列数量,支持1~256。 - 推荐: 如果您不知道实际应用场景中请求的平均序列数量是多少,并且显存充足,我们建议设置为256。如果您的应用场景中请求的平均序列数量明显少于256,或者显存资源紧张,我们建议设置为一个略大于平均值的较小值,以进一步降低显存占用,优化服务性能。

初始化时可用的显存比例 - 参数: --gpu-memory-utilization - 用处: 用于控制 FastDeploy 初始化服务的可用显存,默认0.9,即预留10%的显存备用。 - 推荐: 推荐使用0.7。如果服务压测时提示显存不足,可以尝试调低该值。

2.2.2 Chunked Prefill

  • 参数: --max-num-batched-tokens
  • 用处: chunked prefill中限制每个chunk的最大token数量。
  • 推荐: 我们推荐设置为16384,即关闭chunked prefill

2.2.3 可调整的环境变量

Flash Attention 3:FLAGS_flash_attn_version=3 - 描述:开启 Flash Attention 3 算法。该功能仅支持H卡(Hopper架构,如H800)和B卡(Blackwell架构,如B200)等最新一代 NVIDIA GPU。 - 推荐:在支持该功能的硬件上,Flash Attention 3 能带来显著的性能提升且通常不影响模型精度,强烈推荐启用。

拒绝采样:FD_SAMPLING_CLASS=rejection - 描述:拒绝采样即从一个易于采样的提议分布(proposal distribution)中生成样本,避免显式排序从而达到提升采样速度的效果,可以提升推理性能。 - 推荐:这是一种影响效果的较为激进的优化策略,我们还在全面验证影响。如果对性能有较高要求,也可以接受对效果的影响时可以尝试开启。

三、常见问题FAQ

3.1 显存不足(OOM)

如果服务启动时提示显存不足,请尝试以下方法: 1. 确保无其他进程占用显卡显存; 2. 酌情降低上下文长度和最大序列数量。

如果可以服务可以正常启动,运行时提示显存不足,请尝试以下方法: 1. 酌情降低初始化时可用的显存比例,即调整参数 --gpu-memory-utilization 的值。