Spaces:
Running
on
Zero
Running
on
Zero
adamelliotfields
commited on
Commit
•
c70346f
1
Parent(s):
fd9e8de
Add UniPC scheduler
Browse files- lib/config.py +2 -0
lib/config.py
CHANGED
@@ -10,6 +10,7 @@ from diffusers import (
|
|
10 |
PNDMScheduler,
|
11 |
StableDiffusionImg2ImgPipeline,
|
12 |
StableDiffusionPipeline,
|
|
|
13 |
)
|
14 |
|
15 |
Config = SimpleNamespace(
|
@@ -90,6 +91,7 @@ Config = SimpleNamespace(
|
|
90 |
"Euler": EulerDiscreteScheduler,
|
91 |
"Euler a": EulerAncestralDiscreteScheduler,
|
92 |
"PNDM": PNDMScheduler,
|
|
|
93 |
},
|
94 |
EMBEDDING="fast_negative",
|
95 |
EMBEDDINGS=[
|
|
|
10 |
PNDMScheduler,
|
11 |
StableDiffusionImg2ImgPipeline,
|
12 |
StableDiffusionPipeline,
|
13 |
+
UniPCMultistepScheduler,
|
14 |
)
|
15 |
|
16 |
Config = SimpleNamespace(
|
|
|
91 |
"Euler": EulerDiscreteScheduler,
|
92 |
"Euler a": EulerAncestralDiscreteScheduler,
|
93 |
"PNDM": PNDMScheduler,
|
94 |
+
"UniPC 2M": UniPCMultistepScheduler,
|
95 |
},
|
96 |
EMBEDDING="fast_negative",
|
97 |
EMBEDDINGS=[
|