diff --git a/py/defs/ext/ComfyUI-Lightx02-Nodes.py b/py/defs/ext/ComfyUI-Lightx02-Nodes.py new file mode 100644 index 00000000..d35662b3 --- /dev/null +++ b/py/defs/ext/ComfyUI-Lightx02-Nodes.py @@ -0,0 +1,39 @@ +from ..meta import MetaField + +SAMPLERS = { + "FluxSettingsPipe": { + "positive": "conditioning.positive", + "negative": "conditioning.negative", + }, + "FluxPipeUnpack": { + "positive": "conditioning.positive", + "negative": "conditioning.negative", + }, +} + +CAPTURE_FIELD_LIST = { + "FluxSettingsPipe": { + MetaField.CFG: {"field_name": "cfg"}, + MetaField.SAMPLER_NAME: {"field_name": "sampler_name"}, + MetaField.SCHEDULER: {"field_name": "scheduler"}, + MetaField.STEPS: {"field_name": "steps"}, + MetaField.DENOISE: {"field_name": "denoise"}, + MetaField.SEED: {"field_name": "noise_seed"}, + MetaField.IMAGE_WIDTH: {"field_name": "width"}, + MetaField.IMAGE_HEIGHT: {"field_name": "height"}, + MetaField.POSITIVE_PROMPT: {"field_name": "conditioning.positive"}, + MetaField.NEGATIVE_PROMPT: {"field_name": "conditioning.negative"}, + }, + "FluxPipeUnpack": { + MetaField.CFG: {"field_name": "cfg"}, + MetaField.SAMPLER_NAME: {"field_name": "sampler_name"}, + MetaField.SCHEDULER: {"field_name": "scheduler"}, + MetaField.STEPS: {"field_name": "steps"}, + MetaField.DENOISE: {"field_name": "denoise"}, + MetaField.SEED: {"field_name": "noise_seed"}, + MetaField.IMAGE_WIDTH: {"field_name": "width"}, + MetaField.IMAGE_HEIGHT: {"field_name": "height"}, + MetaField.POSITIVE_PROMPT: {"field_name": "conditioning.positive"}, + MetaField.NEGATIVE_PROMPT: {"field_name": "conditioning.negative"}, + }, +}