Clybius
52eac1b7c8
Add RES step method to Supreme Add spectral noise modulation to Supreme Change reversible dampen to reversible eta on Supreme Remove dyneta temporarily(?) from Supreme Add weight scaling to image/tonal guidance nodes TODO: Update Readme, add start/stop for image guidance, changeable warmup on Supreme (tomorrow)
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from . import extra_samplers
|
|
from . import nodes
|
|
|
|
extra_samplers.add_samplers()
|
|
#extra_samplers.add_schedulers()
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
## K-Samplers
|
|
"SamplerCustomNoise": nodes.SamplerCustomNoise,
|
|
"SamplerCustomNoiseDuo": nodes.SamplerCustomNoiseDuo,
|
|
"SamplerCustomModelMixtureDuo": nodes.SamplerCustomModelMixtureDuo,
|
|
# Guiders
|
|
"GeometricCFGGuider": nodes.GeometricCFGGuider,
|
|
"ImageAssistedCFGGuider": nodes.ImageGuidedCFGGuider,
|
|
"ScaledCFGGuider": nodes.ScaledCFGGuider,
|
|
"WarmupDecayCFGGuider": nodes.WarmupDecayCFGGuider,
|
|
"MegaCFGGuider": nodes.MegaCFGGuider,
|
|
## Samplers
|
|
"SamplerRES_Momentumized": nodes.SamplerRES_MOMENTUMIZED,
|
|
"SamplerDPMPP_DualSDE_Momentumized": nodes.SamplerDPMPP_DUALSDE_MOMENTUMIZED,
|
|
"SamplerCLYB_4M_SDE_Momentumized": nodes.SamplerCLYB_4M_SDE_MOMENTUMIZED,
|
|
"SamplerTTM": nodes.SamplerTTM,
|
|
"SamplerLCMCustom": nodes.SamplerLCMCustom,
|
|
"SamplerEulerAncestralDancing_Experimental": nodes.SamplerEULER_ANCESTRAL_DANCING,
|
|
"SamplerDPMPP_3M_SDE_DynETA": nodes.SamplerDPMPP_3M_SDE_DYN_ETA,
|
|
"SamplerSupreme": nodes.SamplerSUPREME,
|
|
### Schedulers
|
|
"SimpleExponentialScheduler": nodes.SimpleExponentialScheduler,
|
|
}
|
|
__all__ = ['NODE_CLASS_MAPPINGS']
|