Fioceen commited on
Commit
5dc71aa
·
1 Parent(s): 1778b43

Add LUT toggle to ComfyUI Node

Browse files
Files changed (1) hide show
  1. nodes.py +6 -1
nodes.py CHANGED
@@ -87,6 +87,8 @@ class NovaNodes:
87
  "read_noise": ("FLOAT", {"default": 2.0, "min": 0.0, "max": 50.0, "step": 0.1}),
88
 
89
  # LUT
 
 
90
  "lut": ("STRING", {"default": "X://insert/path/here.npy", "vhs_path_extensions": lut_extensions}),
91
  "lut_strength": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}),
92
  },
@@ -131,6 +133,8 @@ class NovaNodes:
131
  enable_bayer=True,
132
  iso_scale=1.0,
133
  read_noise=2.0,
 
 
134
  lut="",
135
  lut_strength=1.0):
136
 
@@ -226,7 +230,8 @@ class NovaNodes:
226
  read_noise=read_noise,
227
  seed=None,
228
  cutoff=0.25,
229
- lut=(lut if lut != "" else None),
 
230
  lut_strength=lut_strength,
231
  )
232
 
 
87
  "read_noise": ("FLOAT", {"default": 2.0, "min": 0.0, "max": 50.0, "step": 0.1}),
88
 
89
  # LUT
90
+ # --- CHANGE 1: Added enable_lut toggle ---
91
+ "enable_lut": ("BOOLEAN", {"default": False}),
92
  "lut": ("STRING", {"default": "X://insert/path/here.npy", "vhs_path_extensions": lut_extensions}),
93
  "lut_strength": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}),
94
  },
 
133
  enable_bayer=True,
134
  iso_scale=1.0,
135
  read_noise=2.0,
136
+ # --- CHANGE 2: Added enable_lut to the function signature ---
137
+ enable_lut=False,
138
  lut="",
139
  lut_strength=1.0):
140
 
 
230
  read_noise=read_noise,
231
  seed=None,
232
  cutoff=0.25,
233
+ # --- CHANGE 3: Updated logic to check enable_lut toggle ---
234
+ lut=(lut if enable_lut and lut != "" else None),
235
  lut_strength=lut_strength,
236
  )
237