kadirnar commited on
Commit
fe8561f
1 Parent(s): 485bc76

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")