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

Co-authored-by: Mick <mickjagger19@icloud.com>
2026-03-24 15:17:33 +03:00
parent 4fbb311234
commit eefb504f84
1 changed files with 3 additions and 1 deletions
@@ -38,7 +38,9 @@ def prepare_mu(batch: Req, server_args: ServerArgs):
    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(
        image_seq_len,