44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
# Copyright (C) 2024 Reece H. Dunn. SPDX-License-Identifier: GPL-3
|
|
|
|
|
|
# noinspection PyPep8Naming,PyMethodMayBeStatic,PyRedundantParentheses,PyMethodParameters
|
|
class LatentImageBusNode:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {"latent": ("LATENT", ),
|
|
"vae": ("VAE", )},
|
|
}
|
|
|
|
RETURN_TYPES = ("LATENT", "VAE", )
|
|
RETURN_NAMES = ("latent", "vae", )
|
|
|
|
FUNCTION = "bus"
|
|
CATEGORY = "ComfyBus"
|
|
|
|
def bus(self, latent, vae):
|
|
return (latent, vae)
|
|
|
|
|
|
# noinspection PyPep8Naming,PyMethodMayBeStatic,PyRedundantParentheses,PyMethodParameters
|
|
class ImageBusNode:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {"image": ("IMAGE", ),
|
|
"mask": ("MASK", )},
|
|
"optional": {"vae": ("VAE", )}}
|
|
|
|
RETURN_TYPES = ("IMAGE", "MASK", "VAE", )
|
|
RETURN_NAMES = ("image", "mask", "vae", )
|
|
|
|
FUNCTION = "bus"
|
|
CATEGORY = "ComfyBus"
|
|
|
|
def bus(self, image, mask, vae = None):
|
|
return (image, mask, vae)
|