执行摘要
本次PR修复了SGLang中FLUX.1和FLUX.2扩散模型的bug，通过启用guidance_embeds=True加载检查点权重并移除guidance_scale的错误缩放，对齐HuggingFace Diffusers实现。影响范围限于多模态生成模块，提升了生成准确性和一致性，建议团队关注条件缩放逻辑的设计权衡。
功能与动机
PR body明确指出：FLUX.1-dev和FLUX.2-dev检查点包含训练好的guidance_embedder权重，但SGLang在guidance_embeds=False时丢弃这些权重。设置guidance_embeds=True是正确的修复，然而_build_guidance()函数错误地将guidance_scale乘以1000，导致模型的正弦时间步嵌入层接收分布外的输入。Diffusers直接传递原始guidance_scale值，因此需要移除缩放以匹配其规范。
实现拆解
- 编码器配置模块：新增
flux_2.py定义FLUX.2的Mistral文本编码器配置，包括build_flux2_text_messages函数和Flux2MistralTextConfig类；在flux.py中移除冗余的Flux2MistralTextArchConfig和format_text_input，改用build_flux2_text_messages。
- 采样参数模块：更新
flux.py中的sampling params，修正默认值：FluxSamplingParams.guidance_scale=3.5，新增Flux2SamplingParams.guidance_scale=4.0；在registry.py中注册配置时使用Flux2SamplingParams。
-
运行时逻辑模块：在denoising.py中修改_build_guidance()函数，通过判断pipeline_config类型，仅对FLUX.1模型保持*1000缩放：
if isinstance(self.server_args.pipeline_config, FluxPipelineConfig) and not isinstance(self.server_args.pipeline_config, Flux2PipelineConfig):
guidance_val = guidance_val * 1000.0
在text_encoding.py中调整attention_mask处理，区分FLUX.1和FLUX.2的编码流程。
-
加载器模块：在component_loader.py中将Flux2PipelineConfig的tokenizer加载从AutoTokenizer改为AutoProcessor以对齐Diffusers。
- 测试模块：在
test_sampling_params.py中添加测试验证guidance_scale默认值匹配模型需求。
评论区精华
review中仅有gemini-code-assist[bot]的一条评论：
"This pull request removes the 1000.0 scaling factor from the guidance tensor construction in the denoising stage to align with the HuggingFace Diffusers convention and prevent out-of-distribution embeddings."
无其他讨论，变更直接通过，表明团队对对齐Diffusers的共识。
风险与影响
- 风险：
1) 回归风险：修改默认值和缩放逻辑可能影响FLUX模型生成质量，需准确性测试保障；
2) 兼容性：条件缩放逻辑需精确区分FLUX.1/FLUX.2，否则可能导致嵌入计算错误；
3) 测试覆盖：新增配置和逻辑变更缺乏充分单元测试，可能隐藏bug。
- 影响：范围限于扩散模型生成管线，程度中等。修正后提升生成准确性，确保与Diffusers一致性，对用户提供更可靠图像生成，对团队展示代码重构和对齐最佳实践。
关联脉络
从近期历史PR看，
-
22157 "[CI] No diffusers backend in lora case"：同属扩散模型修复，涉及CI测试和与Diffusers的一致性，共享多模态生成模块。
-
22146 "Isolate spec V1 path in decode post-processing"：类似结构重构模式，隔离不同版本路径，与本次PR中区分FLUX.1/FLUX.2逻辑有共通设计思路，反映仓库在演进中注重版本管理和一致性维护。
参与讨论