Spaces:
Paused
Paused
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 |
|