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

Documentation

• Class name: TwoSamplersForMask
• Category: ImpactPack/Sampler
• Output node: False
• Repo Ref: https://github.com/ltdrdata/ComfyUI-Impact-Pack.git

该节点通过整合两种不同的采样机制来协调采样过程，允许生成符合特定遮罩标准的潜在表示。它增强了模型关注相关特征和抑制不需要的信息的能力，从而实现更受控和针对性的采样结果。

Input types

Required

• latent_image
  • 潜在图像是携带采样过程所需基础结构和特征的输入表示。它至关重要，因为它构成了节点操作的基础，影响最终输出的质量和特征。
  • Comfy dtype: LATENT
  • Python dtype: dict
• base_sampler
  • 基础采样器是提供基础采样能力的重要组件。它负责初步生成潜在图像，为掩码采样器进一步的细化和操作奠定基础。
  • Comfy dtype: KSAMPLER
  • Python dtype: KSamplerWrapper or KSamplerAdvancedWrapper
• mask_sampler
  • 掩码采样器在应用特定遮罩标准到潜在图像中扮演关键角色。它通过专注于所需特征和抑制不相关特征来精炼采样过程，从而根据预定义的掩码塑造最终输出。
  • Comfy dtype: KSAMPLER
  • Python dtype: KSamplerWrapper or KSamplerAdvancedWrapper
• mask
  • 掩码参数是一个二进制表示，它决定了应该保留或丢弃潜在图像的哪些区域。它是节点操作的一个组成部分，因为它指导掩码采样器确定应该关注哪些特征以及忽略哪些特征。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Output types

• latent_image
  • 输出潜在图像是结合了基础和掩码采样器结果的精炼表示。它根据掩码封装了所需的特征并抑制了不相关的特征，作为后续模型操作的最终输入。
  • Comfy dtype: LATENT
  • Python dtype: dict

Usage tips

• Infra type: GPU

Source code

class TwoSamplersForMask:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'latent_image': ('LATENT',), 'base_sampler': ('KSAMPLER',), 'mask_sampler': ('KSAMPLER',), 'mask': ('MASK',)}}
    RETURN_TYPES = ('LATENT',)
    FUNCTION = 'doit'
    CATEGORY = 'ImpactPack/Sampler'

    def doit(self, latent_image, base_sampler, mask_sampler, mask):
        inv_mask = torch.where(mask != 1.0, torch.tensor(1.0), torch.tensor(0.0))
        latent_image['noise_mask'] = inv_mask
        new_latent_image = base_sampler.sample(latent_image)
        new_latent_image['noise_mask'] = mask
        new_latent_image = mask_sampler.sample(new_latent_image)
        del new_latent_image['noise_mask']
        return (new_latent_image,)