nalin0503 commited on
Commit
7d6b416
·
1 Parent(s): b5e1a79

Fix "LCM Multi-Step Scheduler" display

Browse files
Files changed (1) hide show
  1. 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)