diffusers

Add offload option in flux-control training

#10225

Merged