Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
c4a2b6e
1
Parent(s):
486291a
change base model
Browse files
app_haircolor_inpaint_15.py
CHANGED
|
@@ -14,6 +14,7 @@ from diffusers import (
|
|
| 14 |
ControlNetModel,
|
| 15 |
DDIMScheduler,
|
| 16 |
DPMSolverMultistepScheduler,
|
|
|
|
| 17 |
)
|
| 18 |
|
| 19 |
from controlnet_aux import (
|
|
@@ -61,7 +62,7 @@ basepipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
|
| 61 |
controlnet=controlnet,
|
| 62 |
)
|
| 63 |
# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
|
| 64 |
-
basepipeline.scheduler =
|
| 65 |
|
| 66 |
basepipeline = basepipeline.to(DEVICE)
|
| 67 |
|
|
|
|
| 14 |
ControlNetModel,
|
| 15 |
DDIMScheduler,
|
| 16 |
DPMSolverMultistepScheduler,
|
| 17 |
+
EulerAncestralDiscreteScheduler,
|
| 18 |
)
|
| 19 |
|
| 20 |
from controlnet_aux import (
|
|
|
|
| 62 |
controlnet=controlnet,
|
| 63 |
)
|
| 64 |
# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
|
| 65 |
+
basepipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(basepipeline.scheduler.config)
|
| 66 |
|
| 67 |
basepipeline = basepipeline.to(DEVICE)
|
| 68 |
|