Skip to content

Commit 1e0d2c3

Browse files
authored
Add EulerADiscreteScheduler to utils.py (#634)
1 parent 424a99b commit 1e0d2c3

File tree

1 file changed

+5
-0
lines changed
  • models/turbine_models/custom_models/sd_inference

1 file changed

+5
-0
lines changed

models/turbine_models/custom_models/sd_inference/utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from diffusers import (
77
PNDMScheduler,
88
EulerDiscreteScheduler,
9+
EulerAncestralDiscreteScheduler,
910
)
1011

1112
# If flags are verified to work on a specific model and improve performance without regressing numerics, add them to this dictionary. If you are working with bleeding edge flags, please add them manually with the --ireec_flags argument.
@@ -211,4 +212,8 @@ def get_schedulers(model_id):
211212
model_id,
212213
subfolder="scheduler",
213214
)
215+
schedulers["EulerA"] = EulerAncestralDiscreteScheduler.from_pretrained(
216+
model_id,
217+
subfolder="scheduler",
218+
)
214219
return schedulers

0 commit comments

Comments
 (0)