30 lines
1.5 KiB
Python
30 lines
1.5 KiB
Python
from .nodes.node import *
|
|
from .nodes.random import *
|
|
from .nodes.similar_resolution import *
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SizeFromPresetsSD15": SizeFromPresetsSD15,
|
|
"SizeFromPresetsSDXL": SizeFromPresetsSDXL,
|
|
"EmptyLatentImageFromPresetsSD15": EmptyLatentImageFromPresetsSD15,
|
|
"EmptyLatentImageFromPresetsSDXL": EmptyLatentImageFromPresetsSDXL,
|
|
"RandomSizeFromPresetsSD15": RandomSizeFromPresetsSD15,
|
|
"RandomSizeFromPresetsSDXL": RandomSizeFromPresetsSDXL,
|
|
"RandomEmptyLatentImageFromPresetsSD15": RandomEmptyLatentImageFromPresetsSD15,
|
|
"RandomEmptyLatentImageFromPresetsSDXL": RandomEmptyLatentImageFromPresetsSDXL,
|
|
"GetSimilarResolution": GetSimilarResolution,
|
|
"GetSimilarResolutionEmptyLatent": GetSimilarResolutionEmptyLatent,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SizeFromPresetsSD15": "Size From Presets (SD1.5)",
|
|
"SizeFromPresetsSDXL": "Size From Presets (SDXL)",
|
|
"EmptyLatentImageFromPresetsSD15": "Empty Latent Image From Presets (SD1.5)",
|
|
"EmptyLatentImageFromPresetsSDXL": "Empty Latent Image From Presets (SDXL)",
|
|
"RandomSizeFromPresetsSD15": "Random Size From Presets (SD1.5)",
|
|
"RandomSizeFromPresetsSDXL": "Random Size From Presets (SDXL)",
|
|
"RandomEmptyLatentImageFromPresetsSD15": "Random Empty Latent Image From Presets (SD1.5)",
|
|
"RandomEmptyLatentImageFromPresetsSDXL": "Random Empty Latent Image From Presets (SDXL)",
|
|
"GetSimilarResolution": "Get Similar Resolution",
|
|
"GetSimilarResolutionEmptyLatent": "Get Similar Resolution Empty Latent",
|
|
}
|