ByteDance
/

SDXL-Lightning

stable-diffusion

Model card Files Files and versions

PeterL1n commited on Feb 21, 2024

Commit

397f5c2

·

verified ·

1 Parent(s): 1dcaccf

Update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -77,7 +77,7 @@ from huggingface_hub import hf_hub_download
 base = "stabilityai/stable-diffusion-xl-base-1.0"
 repo = "ByteDance/SDXL-Lightning"
-ckpt = "sdxl_lightning_1step_unet.pth" # Use the correct ckpt for your step setting!
 # Load model.
 pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")

 base = "stabilityai/stable-diffusion-xl-base-1.0"
 repo = "ByteDance/SDXL-Lightning"
+ckpt = "sdxl_lightning_1step_unet_x0.pth" # Use the correct ckpt for your step setting!
 # Load model.
 pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")