| from .channel_select import ChannelSelector
|
| from .mask_image import MaskImage
|
| from .save_surreal import SaveJsonToSurreal, SaveTextToSurreal
|
| from .fetch_surreal import FetchJsonFromSurreal
|
| from .foreground_mask import ForegroundMask
|
| from .save_to_s3 import SaveImageToS3
|
| from .redis import SaveToRedis, FetchFromRedis
|
| from .fal import FalDifferentialDiffusion, FalDiffusion
|
| from .background_select import BackgroundSelect
|
| from .layer_mask import GetLayerMask
|
| from .stream import SendImageOnWebSocket, SendJsonOnWebSocket
|
| from .separate_mask import SeparateMask
|
| from .face_swap import FaceSwap
|
|
|
| NODE_CLASS_MAPPINGS = {
|
| "ChannelSelector": ChannelSelector,
|
| "MaskImage": MaskImage,
|
| "SaveImageToS3": SaveImageToS3,
|
| "SaveJsonToSurreal": SaveJsonToSurreal,
|
| "SaveTextToSurreal": SaveTextToSurreal,
|
| "FetchJsonFromSurreal": FetchJsonFromSurreal,
|
| "ForegroundMask": ForegroundMask,
|
| "SaveToRedis": SaveToRedis,
|
| "FetchFromRedis": FetchFromRedis,
|
| "FalDifferentialDiffusion": FalDifferentialDiffusion,
|
| "FalDiffusion": FalDiffusion,
|
| "BackgroundSelect": BackgroundSelect,
|
| "GetLayerMask": GetLayerMask,
|
| "SendImageOnWebSocket": SendImageOnWebSocket,
|
| "SendJsonOnWebSocket": SendJsonOnWebSocket,
|
| "SeparateMask": SeparateMask,
|
| "FaceSwap": FaceSwap
|
| }
|
|
|
|
|
| NODE_DISPLAY_NAME_MAPPINGS = {
|
| "ChannelSelector":"ChannelSelector",
|
| "MaskImage": "MaskImage",
|
| "SaveImageToS3": "SaveImageToS3",
|
| "SaveJsonToSurreal": "SaveJsonToSurreal",
|
| "SaveTextToSurreal": "SaveTextToSurreal",
|
| "FetchJsonFromSurreal": "FetchJsonFromSurreal",
|
| "ForegroundMask": "ForegroundMask",
|
| "SaveToRedis": "SaveToRedis",
|
| "FetchFromRedis": "FetchFromRedis",
|
| "FalDifferentialDiffusion": "FalDifferentialDiffusion",
|
| "FalDiffusion": "FalDiffusion",
|
| "BackgroundSelect": "BackgroundSelect",
|
| "GetLayerMask": "GetLayerMask",
|
| "SendImageOnWebSocket": "SendImageOnWebSocket",
|
| "SendJsonOnWebSocket": "SendJsonOnWebSocket",
|
| "SeparateMask": "SeparateMask",
|
| "FaceSwap": "FaceSwap"
|
| }
|
|
|
