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

Documentation

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

SeargeModelSelector节点简化了选择和组合模型检查点以进行进一步处理的过程。它封装了处理各种检查点类型的复杂性，使用户能够专注于模型的分析和应用，而无需担心底层的技术细节。

Input types

Required

• base_checkpoint
  • base_checkpoint参数至关重要，因为它指定了作为后续操作起点的基础模型检查点。其选择影响下游任务的整体性能和准确性。
  • Comfy dtype: UI.CHECKPOINTS()
  • Python dtype: Union[str, None]
• refiner_checkpoint
  • refiner_checkpoint参数对于提升基础模型的性能至关重要。它引入了完善模型预测的可能性，从而提高了最终输出的质量。
  • Comfy dtype: UI.CHECKPOINTS_WITH_NONE()
  • Python dtype: Union[str, None]
• vae_checkpoint
  • vae_checkpoint参数在工作流程中整合变分自编码器方面起着关键作用。它使得能够整合潜在表示，这对于复杂的数据处理任务可能至关重要。
  • Comfy dtype: UI.VAE_WITH_EMBEDDED()
  • Python dtype: List[str]

Optional

• data
  • data参数作为模型选择过程中可能需要的附加信息的容器。它在确保节点有效和高效运行中起着支持作用。
  • Comfy dtype: SRG_DATA_STREAM
  • Python dtype: Dict[str, Any]

Output types

• data
  • 输出的data参数包含了模型选择过程的结果，包括组合的检查点。它是工作流程中进入后续阶段的关键组成部分，促进了进一步的分析和应用。
  • Comfy dtype: SRG_DATA_STREAM
  • Python dtype: Dict[str, Any]

Usage tips

• Infra type: CPU

Source code

class SeargeModelSelector:

    @classmethod
    def INPUT_TYPES(cls):
        return {'required': {'base_checkpoint': (UI.CHECKPOINTS(),), 'refiner_checkpoint': (UI.CHECKPOINTS_WITH_NONE(),), 'vae_checkpoint': (UI.VAE_WITH_EMBEDDED(),)}, 'optional': {'data': ('SRG_DATA_STREAM',)}}
    RETURN_TYPES = ('SRG_DATA_STREAM',)
    RETURN_NAMES = ('data',)
    FUNCTION = 'get'
    CATEGORY = UI.CATEGORY_UI_INPUTS

    @staticmethod
    def create_dict(base_checkpoint, refiner_checkpoint, vae_checkpoint):
        return {UI.F_BASE_CHECKPOINT: base_checkpoint, UI.F_REFINER_CHECKPOINT: refiner_checkpoint, UI.F_VAE_CHECKPOINT: vae_checkpoint}

    def get(self, base_checkpoint, refiner_checkpoint, vae_checkpoint, data=None):
        if data is None:
            data = {}
        data[UI.S_CHECKPOINTS] = self.create_dict(base_checkpoint, refiner_checkpoint, vae_checkpoint)
        return (data,)