Kevin Gilper
1253bb629e
Preserve exact enhancer OFF MODEL identity and delegate the installed Krea2T text-fusion helper under owned hook cleanup, source pins and wrapper order checks. Keep task models, LoRAs and schedules intact, including Raw removal. Mark inference handles explicitly unversioned, retain adopted target noise masks, and validate native ER-SDE's actual first-sigma adjustment without replacing its solver. Existing versioned node contracts and defaults remain unchanged. Validation: 163 host-free tests, required compile and pinned source arithmetic probes. Installed-host, real tensor/model and visual acceptance remain separate.
72 lines
3.4 KiB
Python
72 lines
3.4 KiB
Python
"""ComfyUI Krea Reference custom nodes.
|
|
|
|
This package registers the Krea 2 V9 and V10 reference-stack nodes and
|
|
the Concept Slider V1 nodes.
|
|
"""
|
|
|
|
from .kg_krea_v9 import KGKrea2ImageGuideCardV9, KGTextEncodeKreaImageReferencesV9
|
|
from .kg_krea_v10 import KGKrea2ImageGuideCardV10, KGTextEncodeKreaImageReferencesV10
|
|
from .kg_krea_slider import KGKrea2ConceptSliderCardV1, KGKrea2ConceptSliderStackV1
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"KGKrea2ImageGuideCardV9": KGKrea2ImageGuideCardV9,
|
|
"KGTextEncodeKreaImageReferencesV9": KGTextEncodeKreaImageReferencesV9,
|
|
"KGKrea2ImageGuideCardV10": KGKrea2ImageGuideCardV10,
|
|
"KGTextEncodeKreaImageReferencesV10": KGTextEncodeKreaImageReferencesV10,
|
|
"KGKrea2ConceptSliderCardV1": KGKrea2ConceptSliderCardV1,
|
|
"KGKrea2ConceptSliderStackV1": KGKrea2ConceptSliderStackV1,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"KGKrea2ImageGuideCardV9": "KG Krea 2 Image Guide Card V9",
|
|
"KGTextEncodeKreaImageReferencesV9": "KG Krea 2 Reference Stack Encoder V9",
|
|
"KGKrea2ImageGuideCardV10": "KG Krea 2 Image Guide Card V10",
|
|
"KGTextEncodeKreaImageReferencesV10": "KG Krea 2 Reference Stack Encoder V10",
|
|
"KGKrea2ConceptSliderCardV1": "KG Krea 2 Concept Slider Card V1",
|
|
"KGKrea2ConceptSliderStackV1": "KG Krea 2 Concept Slider Stack V1",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
|
|
from .kg_krea_v11 import KGKrea2ImageGuideCardV11, KGTextEncodeKreaImageReferencesV11, KGKrea2ConceptSliderCardV11, KGKrea2ConceptSliderStackV11
|
|
|
|
NODE_CLASS_MAPPINGS.update({
|
|
"KGKrea2ImageGuideCardV11": KGKrea2ImageGuideCardV11,
|
|
"KGTextEncodeKreaImageReferencesV11": KGTextEncodeKreaImageReferencesV11,
|
|
"KGKrea2ConceptSliderCardV11": KGKrea2ConceptSliderCardV11,
|
|
"KGKrea2ConceptSliderStackV11": KGKrea2ConceptSliderStackV11,
|
|
})
|
|
NODE_DISPLAY_NAME_MAPPINGS.update({
|
|
"KGKrea2ImageGuideCardV11": "KG Krea 2 Image Guide Card V11",
|
|
"KGTextEncodeKreaImageReferencesV11": "KG Krea 2 Reference Stack Encoder V11",
|
|
"KGKrea2ConceptSliderCardV11": "KG Krea 2 Concept Slider Card V11",
|
|
"KGKrea2ConceptSliderStackV11": "KG Krea 2 Concept Slider Stack V11",
|
|
})
|
|
|
|
from .kg_krea_v12 import (
|
|
KGKrea2PrepareNativeContextV12, KGKrea2NativeAxisV12, KGKrea2CombinedNativeStudiesV12,
|
|
KGKrea2NativePhaseGuiderV12, KGKrea2PrepareOwnedGroundedContextV12, KGKrea2RemoveOwnedReferenceV12,
|
|
KGKrea2NativeEnhancerV12,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS.update({
|
|
"KGKrea2PrepareNativeContextV12": KGKrea2PrepareNativeContextV12,
|
|
"KGKrea2NativeAxisV12": KGKrea2NativeAxisV12,
|
|
"KGKrea2CombinedNativeStudiesV12": KGKrea2CombinedNativeStudiesV12,
|
|
"KGKrea2NativePhaseGuiderV12": KGKrea2NativePhaseGuiderV12,
|
|
"KGKrea2PrepareOwnedGroundedContextV12": KGKrea2PrepareOwnedGroundedContextV12,
|
|
"KGKrea2RemoveOwnedReferenceV12": KGKrea2RemoveOwnedReferenceV12,
|
|
"KGKrea2NativeEnhancerV12": KGKrea2NativeEnhancerV12,
|
|
})
|
|
NODE_DISPLAY_NAME_MAPPINGS.update({
|
|
"KGKrea2PrepareNativeContextV12": "KG Krea 2 Prepare Native Context V12",
|
|
"KGKrea2NativeAxisV12": "KG Krea 2 Native Axis V12",
|
|
"KGKrea2CombinedNativeStudiesV12": "KG Krea 2 Combined Native Studies V12",
|
|
"KGKrea2NativePhaseGuiderV12": "KG Krea 2 Native Phase Guider V12",
|
|
"KGKrea2PrepareOwnedGroundedContextV12": "KG Krea 2 Prepare Owned Grounded Context V12",
|
|
"KGKrea2RemoveOwnedReferenceV12": "KG Krea 2 Remove Owned Reference V12",
|
|
"KGKrea2NativeEnhancerV12": "KG Krea 2 Native Enhancer V12",
|
|
})
|