AlanB
/

lpw_stable_diffusion_mod

AlanB commited on Jan 11, 2023

Commit

271cad0

1 Parent(s): 69e0e4e

Added from diffusers.schedulers import LMSDiscreteScheduler

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -1295,7 +1295,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
         output_type: Optional[str] = "pil",
         save_n_steps: Optional[int] = None,
         **kwargs,):
         batch_size = 1
         if generator == None:

         output_type: Optional[str] = "pil",
         save_n_steps: Optional[int] = None,
         **kwargs,):
+        from diffusers.schedulers import LMSDiscreteScheduler
         batch_size = 1
         if generator == None: