chenchenghao/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

10
python/sglang/multimodal_gen/runtime/models/dits/flux.py
View File

@@ -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)