minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[Diffusion] Fix FLUX.1-schnell time embedding argument mismatch (#17988)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-01-31 11:47:27 +08:00
committed by GitHub
parent ec76c390c9
commit 22aad4e2c4
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/models/dits/flux.py
+5 -5
View File
@@ -541,11 +541,11 @@ class FluxTransformer2DModel(CachableDiT, OffloadableDiTMixin):
) )
hidden_states, _ = self.x_embedder(hidden_states) hidden_states, _ = self.x_embedder(hidden_states)
temb = ( # Only pass guidance to time_text_embed if the model supports it
self.time_text_embed(timestep, pooled_projections) if self.config.guidance_embeds and guidance is not None:
if guidance is None temb = self.time_text_embed(timestep, guidance, pooled_projections)
else 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) encoder_hidden_states, _ = self.context_embedder(encoder_hidden_states)