Spaces:
Sleeping
Sleeping
adamelliotfields
commited on
Commit
•
3338233
1
Parent(s):
0177258
Remove DPM2 schedulers
Browse files- lib/config.py +0 -4
- lib/loader.py +2 -1
lib/config.py
CHANGED
@@ -6,8 +6,6 @@ from diffusers import (
|
|
6 |
DPMSolverMultistepScheduler,
|
7 |
EulerAncestralDiscreteScheduler,
|
8 |
EulerDiscreteScheduler,
|
9 |
-
KDPM2AncestralDiscreteScheduler,
|
10 |
-
KDPM2DiscreteScheduler,
|
11 |
StableDiffusionXLImg2ImgPipeline,
|
12 |
StableDiffusionXLPipeline,
|
13 |
)
|
@@ -40,8 +38,6 @@ Config = SimpleNamespace(
|
|
40 |
"DDIM": DDIMScheduler,
|
41 |
"DEIS 2M": DEISMultistepScheduler,
|
42 |
"DPM++ 2M": DPMSolverMultistepScheduler,
|
43 |
-
"DPM2": KDPM2DiscreteScheduler,
|
44 |
-
"DPM2 a": KDPM2AncestralDiscreteScheduler,
|
45 |
"Euler": EulerDiscreteScheduler,
|
46 |
"Euler a": EulerAncestralDiscreteScheduler,
|
47 |
},
|
|
|
6 |
DPMSolverMultistepScheduler,
|
7 |
EulerAncestralDiscreteScheduler,
|
8 |
EulerDiscreteScheduler,
|
|
|
|
|
9 |
StableDiffusionXLImg2ImgPipeline,
|
10 |
StableDiffusionXLPipeline,
|
11 |
)
|
|
|
38 |
"DDIM": DDIMScheduler,
|
39 |
"DEIS 2M": DEISMultistepScheduler,
|
40 |
"DPM++ 2M": DPMSolverMultistepScheduler,
|
|
|
|
|
41 |
"Euler": EulerDiscreteScheduler,
|
42 |
"Euler a": EulerAncestralDiscreteScheduler,
|
43 |
},
|
lib/loader.py
CHANGED
@@ -174,7 +174,8 @@ class Loader:
|
|
174 |
print(f"{'Enabling' if karras else 'Disabling'} Karras sigmas...")
|
175 |
if not same_scheduler or not same_karras:
|
176 |
self.pipe.scheduler = Config.SCHEDULERS[scheduler](**scheduler_kwargs)
|
177 |
-
self.refiner
|
|
|
178 |
|
179 |
# https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/blob/main/model_index.json
|
180 |
refiner_kwargs = {
|
|
|
174 |
print(f"{'Enabling' if karras else 'Disabling'} Karras sigmas...")
|
175 |
if not same_scheduler or not same_karras:
|
176 |
self.pipe.scheduler = Config.SCHEDULERS[scheduler](**scheduler_kwargs)
|
177 |
+
if self.refiner is not None:
|
178 |
+
self.refiner.scheduler = self.pipe.scheduler
|
179 |
|
180 |
# https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/blob/main/model_index.json
|
181 |
refiner_kwargs = {
|