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