Spaces:
Running
on
Zero
Running
on
Zero
gokaygokay
commited on
Commit
•
5aa2dd1
1
Parent(s):
01cb633
Update app.py
Browse files
app.py
CHANGED
@@ -69,8 +69,8 @@ pipe_cyber.unet.set_attn_processor(AttnProcessor2_0())
|
|
69 |
|
70 |
# Define samplers
|
71 |
samplers = {
|
72 |
-
"Euler a": EulerAncestralDiscreteScheduler.from_config(
|
73 |
-
"DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(
|
74 |
}
|
75 |
|
76 |
DEFAULT_POSITIVE_PREFIX = "score_9, score_8_up, score_7_up, BREAK"
|
|
|
69 |
|
70 |
# Define samplers
|
71 |
samplers = {
|
72 |
+
"Euler a": EulerAncestralDiscreteScheduler.from_config(pipe_pony.scheduler.config),
|
73 |
+
"DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True)
|
74 |
}
|
75 |
|
76 |
DEFAULT_POSITIVE_PREFIX = "score_9, score_8_up, score_7_up, BREAK"
|