54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
import os
|
|
import filecmp
|
|
import shutil
|
|
|
|
import __main__
|
|
|
|
|
|
WEB_DIRECTORY = "js"
|
|
|
|
from .nodes import *
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LatentMirror": LatentMirror,
|
|
"LatentShift": LatentShift,
|
|
"LatentNormalize": LatentNormalize,
|
|
"TSamplerWithTransform": TSamplerWithTransform,
|
|
"TransformSampler": TransformSampler,
|
|
"MirrorTransform": MirrorTransform,
|
|
"ShiftTransform": ShiftTransform,
|
|
"MultiplyTransform": MultiplyTransform,
|
|
"LatentInterpolateTransform": LatentInterpolateTransform,
|
|
"LatentAddTransform": LatentAddTransform,
|
|
"OneTimeMirrorTransform": OneTimeMirrorTransform,
|
|
"OneTimeMultiplyTransform": OneTimeMultiplyTransform,
|
|
"OneTimeShiftTransform": OneTimeShiftTransform,
|
|
"OneTimeLatentInterpolateTransform": OneTimeLatentInterpolateTransform,
|
|
"OneTimeLatentAddTransform": OneTimeLatentAddTransform,
|
|
"TransformsCombine": TransformsCombine,
|
|
"TransformOffset": TransformOffset,
|
|
"OffsetCombine": OffsetCombine,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LatentMirror": "Latent mirror",
|
|
"LatentShift": "Latent shift",
|
|
"LatentNormalize": "Latent normalize",
|
|
"TSamplerWithTransform": "TSampler with transforms (Latent Control)",
|
|
"TransformSampler": "TSampler (Latent Control)",
|
|
"MirrorTransform": "Mirror transform",
|
|
"ShiftTransform": "Shift transform",
|
|
"MultiplyTransform": "Multiply transform",
|
|
"LatentInterpolateTransform": "Latent interpolate transform",
|
|
"LatentAddTransform": "Latent add transform",
|
|
"OneTimeMirrorTransform": "Mirror transform (one time)",
|
|
"OneTimeMultiplyTransform": "Shift transform (one time)",
|
|
"OneTimeShiftTransform": "Multiply transform (one time)",
|
|
"OneTimeLatentInterpolateTransform": "Latent interpolate transform (one time)",
|
|
"OneTimeLatentAddTransform": "Latent add transform (one time)",
|
|
"TransformsCombine": "Combine transforms",
|
|
"TransformOffset": "Transform offset",
|
|
"OffsetCombine": "Combine offsets",
|
|
}
|
|
|