ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

Nex-N2.5-Pro生产调优清单:从--mamba-scheduler-strategy到--max-running-requests的10个关键参数详解

Nex-N2.5-Pro生产调优清单:从--mamba-scheduler-strategy到--max-running-requests的10个关键参数详解

Nex-N2.5-Pro生产调优清单:从--mamba-scheduler-strategy到--max-running-requests的10个关键参数详解

部署Nex-N2.5-Pro大模型并不是"指向权重就能启动"那么简单——它的混合线性注意力 + MoE 架构对推理引擎有特殊要求。本文基于官方 SGLang 定制镜像的启动命令,整理了一份Nex-N2.5-Pro 生产调优清单,逐个拆解从--mamba-scheduler-strategy到--max-running-requests的 10 个关键参数,帮你把 SGLang 部署一次做对。

先看懂:Nex-N2.5-Pro 为什么需要专门调优 🧠

Nex-N2.5-Pro 是 Nex-AGI 面向长程 Agent 任务的多模态模型,其结构决定了部署参数不能照抄通用模板:

  • 混合注意力架构:60 层中每 4 层为全注意力,其余为线性注意力(Mamba 风格 SSM 状态),见 config.json 中的layer_types与full_attention_interval: 4
  • 512 专家 MoE:每 token 激活 10 个专家(num_experts: 512、num_experts_per_tok: 10)
  • 256K 长上下文:max_position_embeddings: 262144
  • FP8 块量化权重:config.json 中quantization_config采用 FP8_BLOCK 策略,122 个分片由 model.safetensors.index.json 索引

官方推荐配置为单机 8 × H100,使用预置好定制 SGLang 分支的 Docker 镜像nexagi/sglang:v0.5.18-nex-patch。

Nex-N2.5-Pro 完整启动命令

官方推荐的 Pro 版启动命令如下(完整上下文见 README.md):

docker run --gpus all --shm-size 32g --ipc=host \ -p 30000:30000 \ -v /path/to/your/model:/model \ nexagi/sglang:v0.5.18-nex-patch \ python3 -m sglang.launch_server \ --model-path /model \ --tp 8 \ --host 0.0.0.0 --port 30000 \ --reasoning-parser qwen3 \ --tool-call-parser qwen3_coder \ --chat-template /path/to/nex-N2.5-Pro/chat-template.jinja \ --mamba-scheduler-strategy extra_buffer

下面 10 个参数是生产环境的核心,按重要性逐一说明:

#参数推荐值一句话作用
1--mamba-scheduler-strategyextra_buffer混合架构调度的前置条件
2--model-path模型目录定位 122 个权重分片
3--tp8张量并行切分到 8 张卡
4--chat-template仓库模板文件决定对话格式正确性
5--reasoning-parserqwen3分离思考过程与最终回答
6--tool-call-parserqwen3_coder启用函数调用
7--max-running-requests64限制并发解码请求数
8--context-length262144释放 256K 完整上下文
9--mem-fraction-static0.84静态显存占比
10--chunked-prefill-size8192分块预填充,降低首字延迟

1.--mamba-scheduler-strategy extra_buffer:最容易漏掉的一项 ⚠️

这是官方为 Pro / mini 版本强制推荐的开关。由于线性注意力层依赖 Mamba SSM 状态而非传统 KV Cache,SGLang 调度器需要额外的状态缓冲来安全地挂起与恢复长程请求。设置为extra_buffer后,长任务并发下的调度更稳定。生产部署中这一项请务必保留,不要按纯 Transformer 模型的默认值省略。

2.--model-path:指向完整权重目录

指向包含 model-00001-of-00122.safetensors 等 122 个分片及 model.safetensors.index.json、config.json 的目录。Docker 中通过-v挂载到/model,启动时加--trust-remote-code可避免自定义算子加载问题。

3.--tp 8:张量并行是显存的底线

Pro 版权重按 FP8 块量化存储,单机 8 × H100 下--tp 8让每层权重与 KV Cache 均匀切分到 8 张卡。GPU 数量不同时请同步调整,并留意显存是否足够容纳 256K 上下文的缓存。

4.--chat-template:对话格式的正确性开关

指向本仓库的 chat_template.jinja(仓库内文件名为chat_template.jinja,启动命令中按实际文件名传入即可)。模板正确处理图片/视频占位符与思考模式标记,直接决定多模态请求和 Agent 请求能否被正确解析。

5.--reasoning-parser qwen3:把"思考"和"答案"分开

Pro 与 mini 使用qwen3解析器(Max 为deepseek-r1)。它会把模型产生的推理轨迹从最终回答中剥离,客户端即可在流式输出中区分思考内容与可见回复,避免把 CoT 内容直接展示给用户。

6.--tool-call-parser qwen3_coder:函数调用能力

Nex 系列支持稳健的 function calling。加上该参数后,服务端能正确解析工具调用结构,是构建 Agent 工作流的必备开关。

7.--max-running-requests:并发上限怎么定

官方 Max 版示例中使用--max-running-requests 64(见 README.md),Pro 可按显存余量同思路调整。该参数限制同时处于解码阶段的最大请求数:设得太高会加剧显存压力、拉高单请求延迟;设得太低则并发吞吐上不去。经验做法是从 64 起步,观察--mem-fraction-static对应的显存水位与 P99 延迟后再微调。

8.--context-length 262144:释放完整 256K

Pro 的架构上限为 262144(256K)。若业务确需长上下文,请显式设置该参数;若业务只需要 32K,可适当调低以换取更多可缓存容量与更高并发。

9.--mem-fraction-static 0.84:显存静态占比

该参数规定启动时预留给权重与 KV Cache 的显存比例。官方 Max 版示例取0.84,为激活值、CUDA Graph 等运行时开销留出约 16% 余量。若启动即 OOM,先小幅调低此值(如 0.80)定位问题。

10.--chunked-prefill-size 8192+--enable-mixed-chunk:让长输入不再卡死解码

将预填充切成 8192 token 的块,并允许预填充块与解码步混合进同一个 batch。效果是多条长请求同时到达时,首字延迟(TTFT)更平滑、不会长时间饿死短请求——对 Agent 场景这种"长系统提示 + 长上下文"负载尤其重要。

生成质量调优:采样与思考模式 💡

参数调对之后,官方还推荐了一组采样参数(见 README.md):

  • temperature: 0.7
  • top_p: 0.95
  • top_k: 40

思考模式通过reasoning_effort控制:

reasoning_effort模式行为
none非思考直接回答,不产生推理轨迹
medium(默认)自适应思考由模型自行决定是否思考及思考深度
high强制思考回答前始终先思考

⚠️ 注意:该模板使用reasoning_effort字段,enable_thinking、thinking_mode等参数需要网关层做转换。

上线前自查清单 ✅

  1. ☑ 使用nexagi/sglang:v0.5.18-nex-patch镜像,已挂载模型目录
  2. ☑--mamba-scheduler-strategy extra_buffer已加(最关键)
  3. ☑--tp与 GPU 数量一致,--chat-template指向仓库模板
  4. ☑ 推理与工具解析器:qwen3+qwen3_coder
  5. ☑--max-running-requests、--mem-fraction-static按显存水位实测调整
  6. ☑ 客户端采样参数对齐:0.7 / 0.95 / 40

按这份清单逐项核对后,Nex-N2.5-Pro 即可在生产环境稳定提供低延迟、高并发的 Agent 推理服务。更多模型能力与评测数据,可参考仓库 README.md。

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进