examples/flux-control/train_control_flux.py · 745199a86975fb4a1427e20f3ad94070caf0d83c · renzhc / diffusers_dcu

[examples] flux-control: use num_training_steps_for_scheduler (#11662) · 745199a8

Markus Pobitzer authored Jun 05, 2025



[examples] flux-control: use num_training_steps_for_scheduler in get_scheduler instead of args.max_train_steps * accelerator.num_processes
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

745199a8

train_control_flux.py 51.2 KB

Replace train_control_flux.py