kadirnar commited on
Commit
4411eb3
1 Parent(s): a2754d8

Update diffusion_webui/diffusion_models/text2img_app.py

Browse files
diffusion_webui/diffusion_models/text2img_app.py CHANGED
@@ -27,6 +27,7 @@ class StableDiffusionText2ImageGenerator:
27
  self.pipe.to("cuda")
28
  self.pipe.enable_xformers_memory_efficient_attention()
29
  self.pipe.model_name = stable_model_path
 
30
 
31
  return self.pipe
32
 
 
27
  self.pipe.to("cuda")
28
  self.pipe.enable_xformers_memory_efficient_attention()
29
  self.pipe.model_name = stable_model_path
30
+ self.pipe.scheduler_name = scheduler
31
 
32
  return self.pipe
33