Spaces:
Building
on
A10G
Building
on
A10G
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()
|