Repository navigation
Add FLUX.1 export and runtime support - #24376
kiymetakdemir wants to merge 4 commits into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/24376
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (1 Unrelated Failure)As of commit b2f63ac with merge base 893d815 ( BROKEN TRUNK - The following job failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
Summary
Add
black-forest-labs/FLUX.1-devandblack-forest-labs/FLUX.1-schnellsupport to the Diffusers exporter and runner. The pipeline handles CLIP/T5 conditioning, packed image latents, flow-matching schedules and checkpoint-specific guidance defaults through the existing component interfaces.Static and bounded dynamic exports share precomputed RoPE coordinate tables, preserving the reference computation without float64 operations in the exported graph.
Authored with OpenAI Codex.
Test plan
63 tests passed:
python -m pytest export/tests/test_diffusers.py extension/diffusers/test/test_diffusers.py \ -q -k 'not portable_pte and not portable_flux_pte' --disable-warningsExported and ran FLUX.1-dev and FLUX.1-schnell on MLX with a BF16 denoiser and FP32 text encoders/VAE. Checked dynamic execution at 768x1024. Dev's static and dynamic programs produced pixel-identical images at 1024x1024 with the same prompt and seed.