You've already forked ComfyUI-SUPIR
mirror of
https://github.com/kijai/ComfyUI-SUPIR.git
synced 2026-01-27 00:01:30 +03:00
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
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"] |