45 lines
1.7 KiB
Markdown
45 lines
1.7 KiB
Markdown
# 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,)
|
|
``` |