内置节点
简体中文
为Luma图像生成提供参考图像的辅助节点
class LumaReferenceNode(ComfyNodeABC): """ Holds an image and weight for use with Luma Generate Image node. """ RETURN_TYPES = (LumaIO.LUMA_REF,) RETURN_NAMES = ("luma_ref",) DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value FUNCTION = "create_luma_reference" CATEGORY = "api node/image/Luma" @classmethod def INPUT_TYPES(s): return { "required": { "image": ( IO.IMAGE, { "tooltip": "Image to use as reference.", }, ), "weight": ( IO.FLOAT, { "default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01, "tooltip": "Weight of image reference.", }, ), }, "optional": {"luma_ref": (LumaIO.LUMA_REF,)}, } def create_luma_reference( self, image: torch.Tensor, weight: float, luma_ref: LumaReferenceChain = None ): if luma_ref is not None: luma_ref = luma_ref.clone() else: luma_ref = LumaReferenceChain() luma_ref.add(LumaReference(image=image, weight=round(weight, 2))) return (luma_ref,)
此页面对您有帮助吗?