From 22aad4e2c4741b92a37d02e64502c710119cd098 Mon Sep 17 00:00:00 2001 From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Date: Sat, 31 Jan 2026 11:47:27 +0800 Subject: [PATCH] [Diffusion] Fix FLUX.1-schnell time embedding argument mismatch (#17988) --- .../sglang/multimodal_gen/runtime/models/dits/flux.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/sglang/multimodal_gen/runtime/models/dits/flux.py b/python/sglang/multimodal_gen/runtime/models/dits/flux.py index d30c083c0..caf1662de 100644 --- a/python/sglang/multimodal_gen/runtime/models/dits/flux.py +++ b/python/sglang/multimodal_gen/runtime/models/dits/flux.py @@ -541,11 +541,11 @@ class FluxTransformer2DModel(CachableDiT, OffloadableDiTMixin): ) hidden_states, _ = self.x_embedder(hidden_states) - temb = ( - self.time_text_embed(timestep, pooled_projections) - if guidance is None - else self.time_text_embed(timestep, guidance, pooled_projections) - ) + # Only pass guidance to time_text_embed if the model supports it + if self.config.guidance_embeds and guidance is not None: + temb = self.time_text_embed(timestep, guidance, pooled_projections) + else: + temb = self.time_text_embed(timestep, pooled_projections) encoder_hidden_states, _ = self.context_embedder(encoder_hidden_states)