Skip to content

Instantly share code, notes, and snippets.

@alfredplpl

alfredplpl/convert_ckpt_to_diffusers.py

Created November 10, 2023 07:11
Show Gist options
  • Save alfredplpl/b2cd2f99adad328ca0059d446820e476 to your computer and use it in GitHub Desktop.
Save alfredplpl/b2cd2f99adad328ca0059d446820e476 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
convert_ckpt_to_diffusers.py
# MIT
from diffusers import StableDiffusionXLPipeline
import torch
pipe = StableDiffusionXLPipeline.from_single_file('/path/to/checkpoint.safetensors', torch_dtype=torch.float16)
pipe.save_pretrained('/path/to/diffusers_version')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment