You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
553 B
Python
21 lines
553 B
Python
from imaginairy.samplers.base import SolverName # noqa
|
|
from imaginairy.samplers.ddim import DDIMSolver
|
|
|
|
SOLVERS = [
|
|
# PLMSSolver,
|
|
DDIMSolver,
|
|
# kdiff.DPMFastSampler,
|
|
# kdiff.DPMAdaptiveSampler,
|
|
# kdiff.LMSSampler,
|
|
# kdiff.DPM2Sampler,
|
|
# kdiff.DPM2AncestralSampler,
|
|
# kdiff.DPMPP2MSampler,
|
|
# kdiff.DPMPP2SAncestralSampler,
|
|
# kdiff.EulerSampler,
|
|
# kdiff.EulerAncestralSampler,
|
|
# kdiff.HeunSampler,
|
|
]
|
|
|
|
SOLVER_LOOKUP = {s.short_name: s for s in SOLVERS}
|
|
SOLVER_TYPE_OPTIONS = [s.short_name for s in SOLVERS]
|