Clybius
a2ede23da2
Implement IPNDM_VAPP (IPNDM_V with ancestral sampling and CFGPP) Implement STRIKE (A heavily modified Euler with denoised history and 'full ancestral sampling')
38 lines
1.5 KiB
Python
38 lines
1.5 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,
|
|
### Noise
|
|
"ImmiscibleNoise": nodes.ImmiscibleNoise,
|
|
## 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,
|
|
"SamplerSENS": nodes.SamplerSENS,
|
|
"SamplerIPNDM_VAPP": nodes.SamplerIPNDM_VAPP,
|
|
"SamplerSTRIKE": nodes.SamplerSTRIKE,
|
|
### Schedulers
|
|
"SimpleExponentialScheduler": nodes.SimpleExponentialScheduler,
|
|
"KLOptimalScheduler": nodes.KLOptimalScheduler,
|
|
"SimpleKLOptimalScheduler": nodes.SimpleKLOptimalScheduler,
|
|
}
|
|
__all__ = ['NODE_CLASS_MAPPINGS']
|