diffusers

[Training] make checkpointing compatible when using `torch.compile` (part II)

#6511

Merged