67 lines
3.0 KiB
Markdown
67 lines
3.0 KiB
Markdown
# Documentation
|
|
- Class name: DiffusersLoader
|
|
- Category: advanced/loaders/deprecated
|
|
- Output node: False
|
|
- Repo Ref: https://github.com/comfyanonymous/ComfyUI
|
|
|
|
DiffusersLoader 类旨在简化加载和初始化扩散模型的各个组件的过程。它特别适用于从指定路径集成模型本身、CLIP 和 VAE 等不同部分。load_checkpoint 方法是此功能的主要接口，抽象了文件路径解析和模型加载的复杂性。
|
|
|
|
# Input types
|
|
## Required
|
|
- model_path
|
|
- model_path 参数对于 DiffusersLoader 节点至关重要，因为它指定了预期模型文件所在的路径。它用于搜索和加载扩散模型的必需组件。
|
|
- Comfy dtype: str
|
|
- Python dtype: str
|
|
## Optional
|
|
- output_vae
|
|
- output_vae 参数决定是否应与模型一起加载 VAE 组件。它根据应用程序的特定需求提供了模型加载过程中的灵活性。
|
|
- Comfy dtype: bool
|
|
- Python dtype: bool
|
|
- output_clip
|
|
- output_clip 参数控制 CLIP 组件的加载。它允许用户指定 CLIP 模型是否应成为加载的模型组件的一部分。
|
|
- Comfy dtype: bool
|
|
- Python dtype: bool
|
|
|
|
# Output types
|
|
- MODEL
|
|
- MODEL 输出代表了 DiffusersLoader 加载的扩散模型。它是应用程序中进一步处理和分析的中心组件。
|
|
- Comfy dtype: torch.nn.Module
|
|
- Python dtype: torch.nn.Module
|
|
- CLIP
|
|
- CLIP 输出是模型的文本特征提取器部分，它根据 output_clip 参数有条件地加载。它在文本到图像的生成任务中扮演着重要角色。
|
|
- Comfy dtype: torch.nn.Module
|
|
- Python dtype: torch.nn.Module
|
|
- VAE
|
|
- VAE 输出是扩散模型的变分自编码器组件，它根据 output_vae 参数加载。它对于涉及潜在空间操作的任务至关重要。
|
|
- Comfy dtype: comfy.sd.VAE
|
|
- Python dtype: comfy.sd.VAE
|
|
|
|
# Usage tips
|
|
- Infra type: CPU
|
|
|
|
# Source code
|
|
```
|
|
class DiffusersLoader:
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
paths = []
|
|
for search_path in folder_paths.get_folder_paths('diffusers'):
|
|
if os.path.exists(search_path):
|
|
for (root, subdir, files) in os.walk(search_path, followlinks=True):
|
|
if 'model_index.json' in files:
|
|
paths.append(os.path.relpath(root, start=search_path))
|
|
return {'required': {'model_path': (paths,)}}
|
|
RETURN_TYPES = ('MODEL', 'CLIP', 'VAE')
|
|
FUNCTION = 'load_checkpoint'
|
|
CATEGORY = 'advanced/loaders/deprecated'
|
|
|
|
def load_checkpoint(self, model_path, output_vae=True, output_clip=True):
|
|
for search_path in folder_paths.get_folder_paths('diffusers'):
|
|
if os.path.exists(search_path):
|
|
path = os.path.join(search_path, model_path)
|
|
if os.path.exists(path):
|
|
model_path = path
|
|
break
|
|
return comfy.diffusers_load.load_diffusers(model_path, output_vae=output_vae, output_clip=output_clip, embedding_directory=folder_paths.get_folder_paths('embeddings'))
|
|
``` |