KenSchae
/
Pallaidium
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

SDXL LIghtning fix

pull/101/head
tin2tin 9 months ago committed by GitHub
parent
e5614b3129
commit
f4f5bc04cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      __init__.py

2
__init__.py
Unescape View File

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

Write Preview
Loading…
Cancel
Save