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

Documentation

• Class name: PhotoMakerLoader
• Category: _for_testing/photomaker
• Output node: False
• Repo Ref: https://github.com/comfyanonymous/ComfyUI

PhotoMakerLoader 类负责加载和初始化 PhotoMaker 模型，这是图像处理流程中的关键组件。它抽象了模型加载的复杂性，确保模型使用正确的状态字典正确初始化。

Input types

Required

• photomaker_model_name
  • photomaker_model_name 参数对于识别要加载的特定模型至关重要。它指导加载器到正确的文件路径，并确保模型被正确实例化以进行后续操作。
  • Comfy dtype: str
  • Python dtype: str

Output types

• PHOTOMAKER
  • PHOTOMAKER 输出代表已加载的 PhotoMaker 模型，准备用于图像处理任务。它标志着模型与其相关参数和状态的成功初始化。
  • Comfy dtype: PhotoMakerIDEncoder
  • Python dtype: PhotoMakerIDEncoder

Usage tips

• Infra type: CPU

Source code

class PhotoMakerLoader:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'photomaker_model_name': (folder_paths.get_filename_list('photomaker'),)}}
    RETURN_TYPES = ('PHOTOMAKER',)
    FUNCTION = 'load_photomaker_model'
    CATEGORY = '_for_testing/photomaker'

    def load_photomaker_model(self, photomaker_model_name):
        photomaker_model_path = folder_paths.get_full_path('photomaker', photomaker_model_name)
        photomaker_model = PhotoMakerIDEncoder()
        data = comfy.utils.load_torch_file(photomaker_model_path, safe_load=True)
        if 'id_encoder' in data:
            data = data['id_encoder']
        photomaker_model.load_state_dict(data)
        return (photomaker_model,)