Spaces:
Running
on
Zero
Running
on
Zero
foivospar
commited on
Commit
·
b64e72f
1
Parent(s):
cdf0274
add lcm-lora support
Browse files
app.py
CHANGED
|
@@ -58,11 +58,11 @@ pipeline = StableDiffusionPipeline.from_pretrained(
|
|
| 58 |
safety_checker=None
|
| 59 |
)
|
| 60 |
pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
|
| 61 |
-
pipeline = pipeline.to(device)
|
| 62 |
|
| 63 |
# load and disable LCM
|
| 64 |
pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
|
| 65 |
pipeline.disable_lora()
|
|
|
|
| 66 |
|
| 67 |
def toggle_lcm_ui(value):
|
| 68 |
if value:
|
|
|
|
| 58 |
safety_checker=None
|
| 59 |
)
|
| 60 |
pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
|
|
|
|
| 61 |
|
| 62 |
# load and disable LCM
|
| 63 |
pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
|
| 64 |
pipeline.disable_lora()
|
| 65 |
+
pipeline = pipeline.to(device)
|
| 66 |
|
| 67 |
def toggle_lcm_ui(value):
|
| 68 |
if value:
|