Add non_ema checkpoints

No description provided.
This comment has been hidden
patrickvonplaten changed pull request status to open

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:
We made sure you that you can safely merge this pull request.

Best, the 🧨 Diffusers team.

patrickvonplaten changed pull request status to merged

Sign up or log in to comment