File size: 1,239 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 26 27 28 |
from .nodes import SUPIR_Upscale
from .nodes_v2 import SUPIR_sample, SUPIR_model_loader, SUPIR_first_stage, SUPIR_encode, SUPIR_decode, SUPIR_conditioner, SUPIR_tiles, SUPIR_model_loader_v2, SUPIR_model_loader_v2_clip
NODE_CLASS_MAPPINGS = {
"SUPIR_Upscale": SUPIR_Upscale,
"SUPIR_sample": SUPIR_sample,
"SUPIR_model_loader": SUPIR_model_loader,
"SUPIR_first_stage": SUPIR_first_stage,
"SUPIR_encode": SUPIR_encode,
"SUPIR_decode": SUPIR_decode,
"SUPIR_conditioner": SUPIR_conditioner,
"SUPIR_tiles": SUPIR_tiles,
"SUPIR_model_loader_v2": SUPIR_model_loader_v2,
"SUPIR_model_loader_v2_clip": SUPIR_model_loader_v2_clip
}
NODE_DISPLAY_NAME_MAPPINGS = {
"SUPIR_Upscale": "SUPIR Upscale (Legacy)",
"SUPIR_sample": "SUPIR Sampler",
"SUPIR_model_loader": "SUPIR Model Loader (Legacy)",
"SUPIR_first_stage": "SUPIR First Stage (Denoiser)",
"SUPIR_encode": "SUPIR Encode",
"SUPIR_decode": "SUPIR Decode",
"SUPIR_conditioner": "SUPIR Conditioner",
"SUPIR_tiles": "SUPIR Tiles Preview",
"SUPIR_model_loader_v2": "SUPIR Model Loader (v2)",
"SUPIR_model_loader_v2_clip": "SUPIR Model Loader (v2) (Clip)"
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |