Spaces:
Paused
Paused
Update diffusion_webui/diffusion_models/controlnet_pipeline.py
Browse files
diffusion_webui/diffusion_models/controlnet_pipeline.py
CHANGED
@@ -60,7 +60,8 @@ class StableDiffusionControlNetGenerator(ControlnetPipeline):
|
|
60 |
torch_dtype=torch.float16,
|
61 |
)
|
62 |
self.pipe.model_name = stable_model_path
|
63 |
-
|
|
|
64 |
self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
|
65 |
self.pipe.scheduler_name = scheduler
|
66 |
self.pipe.to("cuda")
|
|
|
60 |
torch_dtype=torch.float16,
|
61 |
)
|
62 |
self.pipe.model_name = stable_model_path
|
63 |
+
self.pipe.scheduler_name = scheduler
|
64 |
+
|
65 |
self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
|
66 |
self.pipe.scheduler_name = scheduler
|
67 |
self.pipe.to("cuda")
|