37 lines
1.6 KiB
Python
37 lines
1.6 KiB
Python
from .util import HelperNodes_MultilineStringLiteral, HelperNodes_StringLiteral
|
|
from .util import HelperNodes_Steps, HelperNodes_CfgScale, HelperNodes_WidthHeight
|
|
|
|
from .scheduler import HelperNodes_SchedulerSelector
|
|
|
|
from .sampler import HelperNodes_SamplerSelector, HelperNodes_SeedSelector
|
|
|
|
from .conditioning import HelperNodes_CLIPSkip
|
|
|
|
from .models import HelperNodes_VAESelector, HelperNodes_CheckpointSelector
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"HelperNodes_MultilineStringLiteral": HelperNodes_MultilineStringLiteral,
|
|
"HelperNodes_StringLiteral": HelperNodes_StringLiteral,
|
|
"HelperNodes_Steps": HelperNodes_Steps,
|
|
"HelperNodes_CfgScale": HelperNodes_CfgScale,
|
|
"HelperNodes_WidthHeight": HelperNodes_WidthHeight,
|
|
"HelperNodes_SchedulerSelector": HelperNodes_SchedulerSelector,
|
|
"HelperNodes_SamplerSelector": HelperNodes_SamplerSelector,
|
|
"HelperNodes_SeedSelector": HelperNodes_SeedSelector,
|
|
"HelperNodes_CheckpointSelector": HelperNodes_CheckpointSelector,
|
|
"HelperNodes_VAESelector": HelperNodes_VAESelector,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"HelperNodes_MultilineStringLiteral": "String Literal (multi-line)",
|
|
"HelperNodes_StringLiteral": "String Literal",
|
|
"HelperNodes_Steps": "Steps",
|
|
"HelperNodes_CfgScale": "CFG Scale",
|
|
"HelperNodes_WidthHeight": "Image Dimensions",
|
|
"HelperNodes_SchedulerSelector": "Scheduler Selector",
|
|
"HelperNodes_SamplerSelector": "Sampler Selector",
|
|
"HelperNodes_SeedSelector": "Seed",
|
|
"HelperNodes_CheckpointSelector": "Checkpoint Selector",
|
|
"HelperNodes_VAESelector": "VAE Selector",
|
|
} |