35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
import sys, os
|
|
sys.path.append(os.path.dirname(__file__))
|
|
from .node import (ResetModelPatcherCalculateWeight,
|
|
ApplyLatentInjection,
|
|
VQLoader,
|
|
VQEncoder,
|
|
VQDecoder,
|
|
GetTextMask,
|
|
DetailTransferAdd,
|
|
DetailTransferLatentAdd)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ResetModelPatcherCalculateWeight": ResetModelPatcherCalculateWeight,
|
|
"ApplyLatentInjection": ApplyLatentInjection,
|
|
"VQLoader": VQLoader,
|
|
"VQEncoder": VQEncoder,
|
|
"VQDecoder": VQDecoder,
|
|
"GetTextMask": GetTextMask,
|
|
"DetailTransferAdd":DetailTransferAdd,
|
|
"DetailTransferLatentAdd":DetailTransferLatentAdd,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ResetModelPatcherCalculateWeight": "Reset injected model patcher (middlek)",
|
|
"ApplyLatentInjection": "Apply latent injection (middlek)",
|
|
"VQLoader": "VQ loader diffusers (middlek)",
|
|
"VQEncoder": "VQ encoder diffusers (middlek)",
|
|
"VQDecoder": "VQ decoder diffusers (middlek)",
|
|
"GetTextMask": "Get text mask OCR (middlek)",
|
|
"DetailTransferAdd":"Detail transfer mode:add (middlek)",
|
|
"DetailTransferLatentAdd": "Detail transfer latent mode:add (middlek)",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|