26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
from .node import BatchedMaskedReferenceGen,ReferenceConditionCombine,MasksToPatches,ConditioningMatchMask,GenSubject,PreviewSubjectMask,PreRun,CrossSimMaskExtract,GenConsistent
|
|
NODE_CLASS_MAPPINGS = {
|
|
|
|
"BatchedMaskedReferenceGen": BatchedMaskedReferenceGen,
|
|
"MasksToPatches": MasksToPatches,
|
|
"ReferenceConditionCombine": ReferenceConditionCombine,
|
|
"ConditioningMatchMask": ConditioningMatchMask,
|
|
"GenSubject": GenSubject,
|
|
"PreviewSubjectMask" : PreviewSubjectMask,
|
|
"PreRun": PreRun,
|
|
"CrossSimMaskExtract": CrossSimMaskExtract,
|
|
"GenConsistent": GenConsistent
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ReferenceConditionCombine": "Reference Condition Combine",
|
|
"MasksToPatches": "Masks to Patches",
|
|
"ConditioningMatchMask": "Conditioning Match Mask",
|
|
"BatchedMaskedReferenceGen": "Batched Masked Reference Gen",
|
|
"GenSubject": "Gen Subject",
|
|
"PreviewSubjectMask": "Preview Subject Mask",
|
|
"PreRun": "PreRun",
|
|
"CrossSimMaskExtract": "CrossSim Mask Extract",
|
|
"GenConsistent": "GenConsistent"
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |