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