Documentation

Class name: CheckpointLoaderSimpleMikey
Category: Mikey/Loaders
Output node: False
Repo Ref: https://github.com/bash-j/mikey_nodes

CheckpointLoaderSimpleMikey节点旨在简化机器学习工作流中检查点的加载和管理过程。它提供了一个简单的接口，用于从指定的检查点目录检索模型状态、相关剪辑和变分自编码器（VAE）。该节点还计算并返回检查点文件的哈希值，确保数据的完整性和可追溯性。

Input types

Required

ckpt_name
- 参数'ckpt_name'对于识别要加载的特定检查点至关重要。节点使用它在检查点目录中定位和访问相关的模型状态和配置文件。
- Comfy dtype: str
- Python dtype: str

Optional

output_vae
- 参数'output_vae'决定是否加载与检查点相关联的变分自编码器。它在节点的操作中提供了灵活性，允许用户控制检索检查点的组件。
- Comfy dtype: bool
- Python dtype: bool
output_clip
- 参数'output_clip'控制加载与检查点相关联的clip。它允许节点根据用户的需求选择性地包含或排除clip。
- Comfy dtype: bool
- Python dtype: bool
unique_id
- 参数'unique_id'用于节点操作中的额外识别目的。在某些应用中，可能需要通过文件名之外的方式区分多个检查点，这时'unique_id'就变得相关了。
- Comfy dtype: str
- Python dtype: str
extra_pnginfo
- 参数'extra_pnginfo'旨在为节点内的某些操作提供可能需要的额外信息。它的使用是可选的，并且依赖于上下文，增强了节点对各种场景的适应性。
- Comfy dtype: str
- Python dtype: str
prompt
- 参数'prompt'用于根据用户输入引导节点的行为。它可以影响检查点的处理方式或返回的信息，为节点的执行提供了一定程度的定制化。
- Comfy dtype: str
- Python dtype: str

Output types

model
- 输出'model'代表从检查点加载的机器学习模型状态。它是节点功能的核心组件，使得模型的训练过程可以继续或进行分析。
- Comfy dtype: MODEL
- Python dtype: torch.nn.Module
clip
- 输出'clip'对应于与检查点相关联的加载的clip。它是一个可选组件，可以根据'output_clip'参数包括，为某些应用提供额外的上下文或功能。
- Comfy dtype: CLIP
- Python dtype: torch.nn.Module
vae
- 输出'vae'表示从检查点加载的变分自编码器。它是节点输出的可选部分，由'output_vae'参数控制，可以用于生成任务或进一步分析。
- Comfy dtype: VAE
- Python dtype: torch.nn.Module
ckpt_name
- 输出'ckpt_name'提供了被加载的检查点文件的名称。这对于记录、跟踪或在后续操作中引用特定检查点非常有用。
- Comfy dtype: str
- Python dtype: str
ckpt_hash
- 输出'ckpt_hash'是派生自检查点文件的哈希值。它作为检查点的唯一标识符，对于验证加载数据的完整性或用于记录目的非常有用。
- Comfy dtype: str
- Python dtype: str

Usage tips

Infra type: CPU

Source code

class CheckpointLoaderSimpleMikey:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'ckpt_name': (folder_paths.get_filename_list('checkpoints'),)}, 'hidden': {'unique_id': 'UNIQUE_ID', 'extra_pnginfo': 'EXTRA_PNGINFO', 'prompt': 'PROMPT'}}
    RETURN_TYPES = ('MODEL', 'CLIP', 'VAE', 'STRING', 'STRING')
    RETURN_NAMES = ('model', 'clip', 'vae', 'ckpt_name', 'ckpt_hash')
    FUNCTION = 'load_checkpoint'
    CATEGORY = 'Mikey/Loaders'

    def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True, unique_id=None, extra_pnginfo=None, prompt=None):
        ckpt_path = folder_paths.get_full_path('checkpoints', ckpt_name)
        out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths('embeddings'))
        hash = get_file_hash(ckpt_path)
        ckpt_name = os.path.basename(ckpt_name)
        return out[:3] + (ckpt_name, hash)

4.1 KiB Raw Permalink Blame History

Documentation

Input types

Required

Optional

Output types

Usage tips

Source code

4.1 KiB

Raw Permalink Blame History