执行摘要

• 一句话：撤销FLUX.1-dev ModelOpt NVFP4支持，修复CI测试失败。
• 推荐动作：建议技术管理者精读此PR以理解CI失败原因和revert策略，工程师应关注flux.py中的代码不一致性问题，并考虑后续清理未使用参数。该PR揭示了量化功能集成中的测试和代码一致性挑战。

功能与动机

根据PR body，动机是CI测试失败，具体链接为：failed ci: https://github.com/sgl-project/sglang/actions/runs/24322506533/job/71011288001?pr=22633。作者通过revert提交来快速修复此问题。

实现拆解

实现方案是撤销PR #22574的提交，具体变更包括：

1) 删除文档中的ModelOpt NVFP4支持矩阵（docs/diffusion/quantization.md）；
2) 移除JIT内核预热函数（python/sglang/jit_kernel/nvfp4.py）；
3) 删除NVFP4专用工具脚本（python/sglang/multimodal_gen/tools/build_modelopt_nvfp4_transformer.py）；
4) 重命名并简化FP8工具脚本（python/sglang/multimodal_gen/tools/convert_modelopt_fp8_checkpoint.py）；
5) 修改模型代码以移除prefix参数，但存在不一致性（python/sglang/multimodal_gen/runtime/models/dits/flux.py）；
6) 更新技能文档和测试文件，移除NVFP4相关逻辑。

关键文件：

• docs/diffusion/quantization.md（模块 documentation）: 移除了ModelOpt NVFP4支持矩阵，更新文档以反映功能回退。
• python/sglang/multimodal_gen/tools/build_modelopt_nvfp4_transformer.py（模块 tools）: 完全删除NVFP4专用构建工具脚本，简化量化工作流。
• python/sglang/multimodal_gen/runtime/models/dits/flux.py（模块 models/dits）: 修改FLUX模型代码，移除prefix参数但存在不一致性，review中指出潜在运行时错误。
• python/sglang/multimodal_gen/tools/convert_modelopt_fp8_checkpoint.py（模块 tools）: 重命名并更新FP8工具脚本，从build_modelopt_fp8_transformer.py更名，简化功能聚焦。
• python/sglang/jit_kernel/nvfp4.py（模块 jit_kernel）: 删除prewarm_nvfp4_jit_modules函数，移除JIT内核预热逻辑，影响性能初始化。

关键符号：prewarm_nvfp4_jit_modules, _prepare_nvfp4_weight_bytes, FluxSingleTransformerBlock.init, FluxAttention.init, ModelOptFp4Config.from_config

评论区精华

review评论中，gemini-code-assist[bot]指出两个关键点：

1) 在flux.py中移除prefix参数时不一致，例如to_out和to_add_out的调用仍使用prefix，可能导致运行时错误；
2) prefix参数在FluxSingleTransformerBlock中已未使用，建议后续移除以提高代码清晰度。这些讨论揭示了revert操作可能不完整，存在潜在缺陷。

• 代码不一致性导致潜在运行时错误 (correctness): 建议确保所有ColumnParallelLinear调用一致移除prefix参数，但PR已合并，此问题未解决。
• 未使用参数影响代码清晰度 (style): 建议在后续更改中清理未使用参数，当前PR未处理。

风险与影响

• 风险：技术风险包括：
  1) 代码不一致性：flux.py中prefix参数移除不彻底，可能引发运行时错误（正确性风险）；
  2) 功能移除：撤销NVFP4支持影响用户使用该量化功能，可能破坏依赖此特性的工作流；
  3) 回归风险：revert操作可能意外移除其他必要代码，导致系统行为变化；
  4) 测试覆盖不足：变更涉及多个文件，但review中未提及测试更新是否充分，可能隐藏未发现bug。
• 影响：影响范围：
  1) 用户：无法使用FLUX.1-dev的ModelOpt NVFP4量化功能，性能优化特性暂时缺失；
  2) 系统：恢复至PR #22574前的状态，可能提高稳定性，但损失量化支持；
  3) 团队：需重新评估NVFP4支持的CI失败根因，并规划后续修复或替代方案。影响程度中等，主要限于扩散模型量化领域。
  • 风险标记：代码不一致风险, 功能移除影响, 潜在回归风险

关联脉络

• PR #22574 [Diffusion] Add FLUX.1-dev ModelOpt NVFP4 support: 本PR直接撤销此PR的提交，移除其添加的NVFP4支持功能。
• PR #20082 Enable modelopt quantized FLUX deployment: 同属扩散模型ModelOpt量化功能线，本PR的revert可能影响相关量化部署策略。