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

sayakpaul merged 10 commits intomainhuggingface/diffusers:mainfrom

offload-model-utilityhuggingface/diffusers:offload-model-utilityCopy head branch name to clipboard