Spaces:
Running
Running
zhiweili
commited on
Commit
·
486291a
1
Parent(s):
d36033b
change base model
Browse files
app_haircolor_inpaint_15.py
CHANGED
|
@@ -61,10 +61,7 @@ basepipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
|
| 61 |
controlnet=controlnet,
|
| 62 |
)
|
| 63 |
# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
|
| 64 |
-
basepipeline.scheduler = DPMSolverMultistepScheduler.from_config(
|
| 65 |
-
basepipeline.scheduler.config,
|
| 66 |
-
use_karras_sigmas=True,
|
| 67 |
-
)
|
| 68 |
|
| 69 |
basepipeline = basepipeline.to(DEVICE)
|
| 70 |
|
|
|
|
| 61 |
controlnet=controlnet,
|
| 62 |
)
|
| 63 |
# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
|
| 64 |
+
basepipeline.scheduler = DPMSolverMultistepScheduler.from_config(basepipeline.scheduler.config)
|
|
|
|
|
|
|
|
|
|
| 65 |
|
| 66 |
basepipeline = basepipeline.to(DEVICE)
|
| 67 |
|