Hugging Face's logo

ByteDance
/
SDXL-Lightning

Text-to-Image
Diffusers
stable-diffusion
Model card Files
xet
 Community
63
PeterL1n commited on
Commit
397f5c2
·
verified ·
1 Parent(s): 1dcaccf

Update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -77,7 +77,7 @@ from huggingface_hub import hf_hub_download
77
 
78
  base = "stabilityai/stable-diffusion-xl-base-1.0"
79
  repo = "ByteDance/SDXL-Lightning"
80
- ckpt = "sdxl_lightning_1step_unet.pth" # Use the correct ckpt for your step setting!
81
 
82
  # Load model.
83
  pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")
 
77
 
78
  base = "stabilityai/stable-diffusion-xl-base-1.0"
79
  repo = "ByteDance/SDXL-Lightning"
80
+ ckpt = "sdxl_lightning_1step_unet_x0.pth" # Use the correct ckpt for your step setting!
81
 
82
  # Load model.
83
  pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")