67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
# https://github.com/jags111/efficiency-nodes-comfyui
|
|
from ..meta import MetaField
|
|
from ..formatters import calc_model_hash, convert_skip_clip
|
|
|
|
SAMPLERS = {
|
|
"KSampler (Efficient)": {
|
|
"positive": "positive",
|
|
"negative": "negative",
|
|
},
|
|
"KSampler Adv. (Efficient)": {
|
|
"positive": "positive",
|
|
"negative": "negative",
|
|
},
|
|
"KSampler SDXL (Eff.)": {
|
|
"positive": "positive",
|
|
"negative": "negative",
|
|
},
|
|
}
|
|
|
|
CAPTURE_FIELD_LIST = {
|
|
"Efficient Loader": {
|
|
MetaField.MODEL_NAME: {"field_name": "ckpt_name"},
|
|
MetaField.MODEL_HASH: {"field_name": "ckpt_name", "format": calc_model_hash},
|
|
MetaField.CLIP_SKIP: {"field_name": "clip_skip", "format": convert_skip_clip},
|
|
MetaField.POSITIVE_PROMPT: {"field_name": "positive"},
|
|
MetaField.NEGATIVE_PROMPT: {"field_name": "negative"},
|
|
MetaField.IMAGE_WIDTH: {"field_name": "empty_latent_width"},
|
|
MetaField.IMAGE_HEIGHT: {"field_name": "empty_latent_height"},
|
|
},
|
|
"Eff. Loader SDXL": {
|
|
MetaField.MODEL_NAME: {"field_name": "base_ckpt_name"},
|
|
MetaField.MODEL_HASH: {
|
|
"field_name": "base_ckpt_name",
|
|
"format": calc_model_hash,
|
|
},
|
|
MetaField.CLIP_SKIP: {
|
|
"field_name": "base_clip_skip",
|
|
"format": convert_skip_clip,
|
|
},
|
|
MetaField.POSITIVE_PROMPT: {"field_name": "positive"},
|
|
MetaField.NEGATIVE_PROMPT: {"field_name": "negative"},
|
|
MetaField.IMAGE_WIDTH: {"field_name": "empty_latent_width"},
|
|
MetaField.IMAGE_HEIGHT: {"field_name": "empty_latent_height"},
|
|
},
|
|
"KSampler (Efficient)": {
|
|
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"},
|
|
},
|
|
"KSampler Adv. (Efficient)": {
|
|
MetaField.SEED: {"field_name": "noise_seed"},
|
|
MetaField.STEPS: {"field_name": "steps"},
|
|
MetaField.CFG: {"field_name": "cfg"},
|
|
MetaField.SAMPLER_NAME: {"field_name": "sampler_name"},
|
|
MetaField.SCHEDULER: {"field_name": "scheduler"},
|
|
},
|
|
"KSampler SDXL (Eff.)": {
|
|
MetaField.SEED: {"field_name": "noise_seed"},
|
|
MetaField.STEPS: {"field_name": "steps"},
|
|
MetaField.CFG: {"field_name": "cfg"},
|
|
MetaField.SAMPLER_NAME: {"field_name": "sampler_name"},
|
|
MetaField.SCHEDULER: {"field_name": "scheduler"},
|
|
},
|
|
}
|