Update __init__.py

2026-03-21 21:22:13 -03:00 · 2023-11-30 13:59:17 +05:30
parent 433b1e59fe
commit a3090ee5ec
1 changed files with 102 additions and 2 deletions
--- a/init.py
+++ b/init.py
@@ -1,3 +1,103 @@
+import os
+import subprocess
+import importlib.util
+import folder_paths
+import shutil
+import sys
+import traceback
+
 from  .efficiency_nodes import NODE_CLASS_MAPPINGS
+from  .py.ttl_nn_latent_upscaler import NNLatentUpscale
+from  .py.city96_latent_upscaler import LatentUpscaler
+
+NODE_CLASS_MAPPINGS = {
+    "KSampler (Efficient)": TSC_KSampler,
+    "KSampler Adv. (Efficient)":TSC_KSamplerAdvanced,
+    "KSampler SDXL (Eff.)": TSC_KSamplerSDXL,
+    "Efficient Loader": TSC_EfficientLoader,
+    "Eff. Loader SDXL": TSC_EfficientLoaderSDXL,
+    "LoRA Stacker": TSC_LoRA_Stacker,
+    "Control Net Stacker": TSC_Control_Net_Stacker,
+    "Apply ControlNet Stack": TSC_Apply_ControlNet_Stack,
+    "Unpack SDXL Tuple": TSC_Unpack_SDXL_Tuple,
+    "Pack SDXL Tuple": TSC_Pack_SDXL_Tuple,
+    "XY Plot": TSC_XYplot,
+    "XY Input: Seeds++ Batch": TSC_XYplot_SeedsBatch,
+    "XY Input: Add/Return Noise": TSC_XYplot_AddReturnNoise,
+    "XY Input: Steps": TSC_XYplot_Steps,
+    "XY Input: CFG Scale": TSC_XYplot_CFG,
+    "XY Input: Sampler/Scheduler": TSC_XYplot_Sampler_Scheduler,
+    "XY Input: Denoise": TSC_XYplot_Denoise,
+    "XY Input: VAE": TSC_XYplot_VAE,
+    "XY Input: Prompt S/R": TSC_XYplot_PromptSR,
+    "XY Input: Aesthetic Score": TSC_XYplot_AScore,
+    "XY Input: Refiner On/Off": TSC_XYplot_Refiner_OnOff,
+    "XY Input: Checkpoint": TSC_XYplot_Checkpoint,
+    "XY Input: Clip Skip": TSC_XYplot_ClipSkip,
+    "XY Input: LoRA": TSC_XYplot_LoRA,
+    "XY Input: LoRA Plot": TSC_XYplot_LoRA_Plot,
+    "XY Input: LoRA Stacks": TSC_XYplot_LoRA_Stacks,
+    "XY Input: Control Net": TSC_XYplot_Control_Net,
+    "XY Input: Control Net Plot": TSC_XYplot_Control_Net_Plot,
+    "XY Input: Manual XY Entry": TSC_XYplot_Manual_XY_Entry,
+    "Manual XY Entry Info": TSC_XYplot_Manual_XY_Entry_Info,
+    "Join XY Inputs of Same Type": TSC_XYplot_JoinInputs,
+    "Image Overlay": TSC_ImageOverlay,
+    "Noise Control Script": TSC_Noise_Control_Script,
+    "HighRes-Fix Script": TSC_HighRes_Fix,
+    "Tiled Upscaler Script": TSC_Tiled_Upscaler,
+    "Evaluate Integers": TSC_EvaluateInts,
+    "Evaluate Floats": TSC_EvaluateFloats,
+    "Evaluate Strings": TSC_EvaluateStrs,
+     "Simple Eval Examples": TSC_EvalExamples,
+     "NNLatentUpscale" : NNLatentUpscale,
+     "LatentUpscaler" : LatentUpscaler
+}
+NODE_DISPLAY_NAME_MAPPINGS = {
+     "KSampler (Efficient)": "EFF_KSampler",
+     "KSampler Adv. (Efficient)":"EFF_KSamplerAdvanced",
+     "KSampler SDXL (Eff.)": "EFF_KSamplerSDXL",
+     "Efficient Loader": "EFF_EfficientLoader",
+     "Eff. Loader SDXL": "EFF_EfficientLoaderSDXL",
+     "LoRA Stacker": "EFF_LoRA_Stacker",
+     "Control Net Stacker": "EFF_Control_Net_Stacker",
+     "Apply ControlNet Stack": "EFF_Apply_ControlNet_Stack",
+     "Unpack SDXL Tuple": "EFF_Unpack_SDXL_Tuple",
+     "Pack SDXL Tuple": "EFF_Pack_SDXL_Tuple",
+     "XY Plot": "EFF_XYplot",
+     "XY Input: Seeds++ Batch": "EFF_XYplot_SeedsBatch",
+     "XY Input: Add/Return Noise": "EFF_XYplot_AddReturnNoise",
+     "XY Input: Steps": "EFF_XYplot_Steps",
+     "XY Input: CFG Scale": "EFF_XYplot_CFG",
+     "XY Input: Sampler/Scheduler": "EFF_XYplot_Sampler_Scheduler",
+     "XY Input: Denoise": "EFF_XYplot_Denoise",
+     "XY Input: VAE": "EFF_XYplot_VAE",
+     "XY Input: Prompt S/R": "EFF_XYplot_PromptSR",
+     "XY Input: Aesthetic Score": "EFF_XYplot_AScore",
+     "XY Input: Refiner On/Off": "EFF_XYplot_Refiner_OnOff",
+     "XY Input: Checkpoint": "EFF_XYplot_Checkpoint",
+     "XY Input: Clip Skip": "EFF_XYplot_ClipSkip",
+     "XY Input: LoRA": "EFF_XYplot_LoRA",
+     "XY Input: LoRA Plot": "EFF_XYplot_LoRA_Plot",
+     "XY Input: LoRA Stacks": "EFF_XYplot_LoRA_Stacks",
+     "XY Input: Control Net": "EFF_XYplot_Control_Net",
+     "XY Input: Control Net Plot": "EFF_XYplot_Control_Net_Plot",
+     "XY Input: Manual XY Entry": "EFF_XYplot_Manual_XY_Entry",
+      "Manual XY Entry Info": "EFF_XYplot_Manual_XY_Entry_Info",
+      "Join XY Inputs of Same Type": "EFF_XYplot_JoinInputs",
+      "Image Overlay": "EFF_ImageOverlay",
+      "Noise Control Script": "EFF_Noise_Control_Script",
+      "HighRes-Fix Script": "EFF_HighRes_Fix",
+      "Tiled Upscaler Script": "EFF_Tiled_Upscaler",
+      "Evaluate Integers": "EFF_EvaluateInts",
+      "Evaluate Floats": "EFF_EvaluateFloats",
+      "Evaluate Strings": "EFF_EvaluateStrs",
+      "Simple Eval Examples": "EFF_EvalExamples",
+      "NNLatentUpscale" : "EFF_NNLatentUpscale",
+      "LatentUpscaler" : "EFF_LatentUpscaler"
+}
 WEB_DIRECTORY = "js"
-__all__ = ['NODE_CLASS_MAPPINGS']
+
+CC_VERSION = 2.0
+
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'CC_VERSION']