kadirnar commited on
Commit
6e5ca0e
1 Parent(s): fe8561f

Update diffusion_webui/diffusion_models/controlnet_inpaint_pipeline.py

Browse files
diffusion_webui/diffusion_models/controlnet_inpaint_pipeline.py CHANGED
@@ -37,6 +37,7 @@ class StableDiffusionControlNetInpaintGenerator(ControlnetPipeline):
37
  )
38
 
39
  self.pipe.model_name = stable_model_path
 
40
  self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
41
  self.pipe.to("cuda")
42
  self.pipe.enable_xformers_memory_efficient_attention()
 
37
  )
38
 
39
  self.pipe.model_name = stable_model_path
40
+ self.pipe.scheduler_name = scheduler
41
  self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
42
  self.pipe.to("cuda")
43
  self.pipe.enable_xformers_memory_efficient_attention()