Spaces:
Running
Running
Fix "LCM Multi-Step Scheduler" display
Browse files- Image-Morpher/model.py +4 -0
Image-Morpher/model.py
CHANGED
@@ -516,6 +516,7 @@ class DiffMorpherPipeline(StableDiffusionPipeline):
|
|
516 |
lora_1,
|
517 |
alpha_list[0],
|
518 |
False,
|
|
|
519 |
fix_lora
|
520 |
)
|
521 |
first_image = self.latent2image(latents)
|
@@ -552,6 +553,7 @@ class DiffMorpherPipeline(StableDiffusionPipeline):
|
|
552 |
lora_1,
|
553 |
alpha_list[-1],
|
554 |
False,
|
|
|
555 |
fix_lora
|
556 |
)
|
557 |
last_image = self.latent2image(latents)
|
@@ -592,6 +594,7 @@ class DiffMorpherPipeline(StableDiffusionPipeline):
|
|
592 |
lora_1,
|
593 |
alpha_list[i],
|
594 |
False,
|
|
|
595 |
fix_lora
|
596 |
)
|
597 |
image = self.latent2image(latents)
|
@@ -617,6 +620,7 @@ class DiffMorpherPipeline(StableDiffusionPipeline):
|
|
617 |
lora_1,
|
618 |
alpha_list[k],
|
619 |
self.use_lora,
|
|
|
620 |
fix_lora
|
621 |
)
|
622 |
image = self.latent2image(latents)
|
|
|
516 |
lora_1,
|
517 |
alpha_list[0],
|
518 |
False,
|
519 |
+
self.use_lcm,
|
520 |
fix_lora
|
521 |
)
|
522 |
first_image = self.latent2image(latents)
|
|
|
553 |
lora_1,
|
554 |
alpha_list[-1],
|
555 |
False,
|
556 |
+
self.use_lcm,
|
557 |
fix_lora
|
558 |
)
|
559 |
last_image = self.latent2image(latents)
|
|
|
594 |
lora_1,
|
595 |
alpha_list[i],
|
596 |
False,
|
597 |
+
self.use_lcm,
|
598 |
fix_lora
|
599 |
)
|
600 |
image = self.latent2image(latents)
|
|
|
620 |
lora_1,
|
621 |
alpha_list[k],
|
622 |
self.use_lora,
|
623 |
+
self.use_lcm,
|
624 |
fix_lora
|
625 |
)
|
626 |
image = self.latent2image(latents)
|