CavinHuang-comfyui-nodes-docs/docs/SeargeCustomAfterVaeDecode.md

Documentation

• Class name: SeargeCustomAfterVaeDecode
• Category: UI.CATEGORY_MAGIC_CUSTOM_STAGES
• Output node: False
• Repo Ref: https://github.com/jobunk/SeargeSDXL.git

该节点旨在通过解码和后处理变分自编码器（VAE）的输出来生成最终图像。它封装了将VAE的潜在空间表示转换为视觉输出所需的逻辑，提高了整个工作流程的效率和效果。

Input types

Required

• custom_output
  • custom_output参数至关重要，因为它作为VAE解码过程的输入。它是前一个阶段的输出，包含了节点执行其功能所需的必要信息。
  • Comfy dtype: SRG_STAGE_OUTPUT
  • Python dtype: Dict[str, Any]

Output types

• image
  • 输出图像是VAE解码和后处理的结果。它代表了节点完成操作后的最终视觉产品，对于进一步的分析或显示至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Usage tips

• Infra type: CPU

Source code

class SeargeCustomAfterVaeDecode:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'custom_output': ('SRG_STAGE_OUTPUT',)}, 'optional': {}}
    RETURN_TYPES = ('IMAGE',)
    RETURN_NAMES = ('image',)
    FUNCTION = 'output'
    CATEGORY = UI.CATEGORY_MAGIC_CUSTOM_STAGES

    def output(self, custom_output):
        if custom_output is None:
            return (None,)
        vae_decoded = retrieve_parameter(Names.S_VAE_DECODED, custom_output)
        image = retrieve_parameter(Names.F_DECODED_IMAGE, vae_decoded)
        post_processed = retrieve_parameter(Names.F_POST_PROCESSED, vae_decoded)
        result = image if post_processed is None else post_processed
        return (result,)