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