22 lines
911 B
Python
22 lines
911 B
Python
from .node import BatchedMaskedReferenceGen,ReferenceConditionCombine,MasksToPatches,ConditioningMatchMask,GenSubject,PreviewSubjectMask,PreRun
|
|
NODE_CLASS_MAPPINGS = {
|
|
|
|
"BatchedMaskedReferenceGen": BatchedMaskedReferenceGen,
|
|
"MasksToPatches": MasksToPatches,
|
|
"ReferenceConditionCombine": ReferenceConditionCombine,
|
|
"ConditioningMatchMask": ConditioningMatchMask,
|
|
"GenSubject": GenSubject,
|
|
"PreviewSubjectMask" : PreviewSubjectMask,
|
|
"PreRun": PreRun,
|
|
}
|
|
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",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |