diffusers

set the model to train state before accelerator prepare

#5099

Merged