mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-11-09 13:10:27 +00:00
015088507f
- feature: use different default steps and image sizes depending on sampler and model selceted - fix: #110 use proper version in image metadata - refactor: samplers all have their own class that inherits from ImageSampler
23 lines
657 B
Python
23 lines
657 B
Python
from imaginairy.samplers import kdiff
|
|
from imaginairy.samplers.base import SamplerName # noqa
|
|
from imaginairy.samplers.ddim import DDIMSampler
|
|
from imaginairy.samplers.plms import PLMSSampler
|
|
|
|
SAMPLERS = [
|
|
PLMSSampler,
|
|
DDIMSampler,
|
|
kdiff.DPMFastSampler,
|
|
kdiff.DPMAdaptiveSampler,
|
|
kdiff.LMSSampler,
|
|
kdiff.DPM2Sampler,
|
|
kdiff.DPM2AncestralSampler,
|
|
kdiff.DPMPP2MSampler,
|
|
kdiff.DPMPP2SAncestralSampler,
|
|
kdiff.EulerSampler,
|
|
kdiff.EulerAncestralSampler,
|
|
kdiff.HeunSampler,
|
|
]
|
|
|
|
SAMPLER_LOOKUP = {sampler.short_name: sampler for sampler in SAMPLERS}
|
|
SAMPLER_TYPE_OPTIONS = [sampler.short_name for sampler in SAMPLERS]
|