File size: 1,070 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# from .nodes.unsampler_node import MochiWrapperUnsamplerNode
# from .nodes.resampler_node import MochiWrapperResamplerNode
from .nodes.sampler_custom_node import MochiWrapperSamplerCustomNode
from .nodes.sampler_unsample_node import MochiUnsamplerNode
from .nodes.sampler_resample_node import MochiResamplerNode
from .nodes.prepare_mochi_sigmas_node import MochiPrepareSigmasNode
NODE_CLASS_MAPPINGS = {
# "MochiWrapperUnsampler": MochiWrapperUnsamplerNode,
# "MochiWrapperResampler": MochiWrapperResamplerNode,
"MochiWrapperSamplerCustom": MochiWrapperSamplerCustomNode,
"MochiUnsampler": MochiUnsamplerNode,
"MochiResampler": MochiResamplerNode,
"MochiPrepareSigmas": MochiPrepareSigmasNode,
}
NODE_DISPLAY_NAME_MAPPINGS = {
# "MochiWrapperUnsampler": "Mochi Wrapper Unsampler",
# "MochiWrapperResampler": "Mochi Wrapper Resampler",
"MochiWrapperSamplerCustom": "SamplerCustom (Mochi Wrapper)",
"MochiUnsampler": "Mochi Unsampler",
"MochiResampler": "Mochi Resampler",
"MochiPrepareSigmas": "Mochi Prepare Sigmas",
} |