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

[diffusion] model: Fix FLUX.1 output correctness (#21041)

Browse Source 
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Aleksi Vesanto
2026-03-24 15:17:33 +03:00
committed by GitHub
co-authored by Mick
parent 4fbb311234
commit eefb504f84
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/pipelines/flux.py
+3 -1
View File
@@ -38,7 +38,9 @@ def prepare_mu(batch: Req, server_args: ServerArgs):
vae_scale_factor = ( vae_scale_factor = (
server_args.pipeline_config.vae_config.arch_config.vae_scale_factor server_args.pipeline_config.vae_config.arch_config.vae_scale_factor
) )
image_seq_len = (int(height) // vae_scale_factor) * (int(width) // vae_scale_factor) image_seq_len = (int(height) // (vae_scale_factor * 2)) * (
int(width) // (vae_scale_factor * 2)
)
mu = calculate_shift( mu = calculate_shift(
image_seq_len, image_seq_len,