Add non_ema checkpoints
Hey CompVis 👋,
Your model repository seems to contain a non-ema branch to load the unet non-ema checkpoint. Loading non-ema versions from a branch instead of the main branch is deprecated and will eventually be forbidden. Instead, we strongly recommend to save non-ema versions of the model under .non-ema. version files directly on the 'main' branch as enabled through this PR.This PR makes sure that your model repository allows the user to correctly download non-ema unet model weights by adding non-ema unet model weights in both safetensors and PyTorch bin format:
pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16, variant='non_ema')
For more information please have a look at: https://huggingface.co/docs/diffusers/using-diffusers/loading#checkpoint-variants.
We made sure you that you can safely merge this pull request.
Best, the 🧨 Diffusers team.