The original checkpoint saved just the transformer model but in the diffusers checkpoint the transformer is nested in the model and diffusion_model. It seems that the transformer folder would be loadable as is but this small addition to sanitize() will work just as well.
Fixes #1116 .
The original checkpoint saved just the transformer model but in the diffusers checkpoint the transformer is nested in the
modelanddiffusion_model. It seems that the transformer folder would be loadable as is but this small addition tosanitize()will work just as well.