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

Documentation

• Class name: InstantIDModelLoader
• Category: InstantID
• Output node: False
• Repo Ref: https://github.com/cubiq/ComfyUI_InstantID.git

该节点旨在高效加载和管理InstantID特定模型，确保检索并准备适当的模型以供系统内使用。

Input types

Required

• instantid_file
  • 该参数对于指定要加载的确切InstantID模型文件至关重要。它决定了后续操作中将使用模型的身份和版本。
  • Comfy dtype: STRING
  • Python dtype: str

Output types

• INSTANTID
  • 输出代表加载的InstantID模型，这对于系统执行与模型功能相关的任务至关重要。
  • Comfy dtype: DICTIONARY
  • Python dtype: Dict[str, Any]

Usage tips

• Infra type: CPU

Source code

class InstantIDModelLoader:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'instantid_file': (folder_paths.get_filename_list('instantid'),)}}
    RETURN_TYPES = ('INSTANTID',)
    FUNCTION = 'load_model'
    CATEGORY = 'InstantID'

    def load_model(self, instantid_file):
        ckpt_path = folder_paths.get_full_path('instantid', instantid_file)
        model = comfy.utils.load_torch_file(ckpt_path, safe_load=True)
        if ckpt_path.lower().endswith('.safetensors'):
            st_model = {'image_proj': {}, 'ip_adapter': {}}
            for key in model.keys():
                if key.startswith('image_proj.'):
                    st_model['image_proj'][key.replace('image_proj.', '')] = model[key]
                elif key.startswith('ip_adapter.'):
                    st_model['ip_adapter'][key.replace('ip_adapter.', '')] = model[key]
            model = st_model
        return (model,)