diffusers

[training] add an offload utility that can be used as a context manager.

#11775

Merged