Spaces:
Runtime error
Runtime error
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()
|