diff --git a/py/defs/captures.py b/py/defs/captures.py index ef04ef2..8f03ec1 100644 --- a/py/defs/captures.py +++ b/py/defs/captures.py @@ -29,4 +29,11 @@ CAPTURE_FIELD_LIST = { MetaField.SAMPLER_NAME: {"field_name": "sampler_name"}, MetaField.SCHEDULER: {"field_name": "scheduler"}, }, + "KSamplerAdvanced": { + MetaField.SEED: {"field_name": "seed"}, + MetaField.STEPS: {"field_name": "steps"}, + MetaField.CFG: {"field_name": "cfg"}, + MetaField.SAMPLER_NAME: {"field_name": "sampler_name"}, + MetaField.SCHEDULER: {"field_name": "scheduler"}, + }, } diff --git a/py/defs/samplers.py b/py/defs/samplers.py index 6a84790..60e3530 100644 --- a/py/defs/samplers.py +++ b/py/defs/samplers.py @@ -3,4 +3,8 @@ SAMPLERS = { "positive": "positive", "negative": "negative", }, + "KSamplerAdvanced": { + "positive": "positive", + "negative": "negative", + }, }