Spaces:
Running
Running
Update diffusion_webui/diffusion_models/inpaint_app.py
Browse files
diffusion_webui/diffusion_models/inpaint_app.py
CHANGED
@@ -16,7 +16,9 @@ class StableDiffusionInpaintGenerator:
|
|
16 |
)
|
17 |
self.pipe.to("cuda")
|
18 |
self.pipe.enable_xformers_memory_efficient_attention()
|
19 |
-
|
|
|
|
|
20 |
return self.pipe
|
21 |
|
22 |
def generate_image(
|
|
|
16 |
)
|
17 |
self.pipe.to("cuda")
|
18 |
self.pipe.enable_xformers_memory_efficient_attention()
|
19 |
+
self.pipe.scheduler_name = scheduler
|
20 |
+
self.pipe.model_name = stable_model_path
|
21 |
+
|
22 |
return self.pipe
|
23 |
|
24 |
def generate_image(
|