28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
from .node import BatchedMaskedReferenceGen,ReferenceConditionCombine,MasksToPatches,ConditioningMatchMask,GenSubject,PreviewSubjectMask,PreRun,CrossSimMaskExtract,GenConsistent,SetSubjectMask
|
|
NODE_CLASS_MAPPINGS = {
|
|
|
|
"BatchedMaskedReferenceGen": BatchedMaskedReferenceGen,
|
|
"MasksToPatches": MasksToPatches,
|
|
"ReferenceConditionCombine": ReferenceConditionCombine,
|
|
"ConditioningMatchMask": ConditioningMatchMask,
|
|
"GenSubject": GenSubject,
|
|
"PreviewSubjectMask" : PreviewSubjectMask,
|
|
"PreRun": PreRun,
|
|
"CrossSimMaskExtract": CrossSimMaskExtract,
|
|
"GenConsistent": GenConsistent,
|
|
"SetSubjectMask": SetSubjectMask,
|
|
}
|
|
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",
|
|
"SetSubjectMask": "SetSubjectMask"
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |