diffusers

[advanced flux training] bug fix + reduce memory cost as in #9829

#9838

Merged