65 lines
2.1 KiB
Markdown
65 lines
2.1 KiB
Markdown
|
|
# Documentation
|
|
- Class name: easy fooocusInpaintLoader
|
|
- Category: EasyUse/Inpaint
|
|
- Output node: False
|
|
- Repo Ref: https://github.com/comfyanonymous/ComfyUI
|
|
|
|
easy fooocusInpaintLoader节点旨在简化ComfyUI框架中修复模型和补丁的加载与应用过程。它主要专注于将修复功能整合到模型处理流程中，通过应用学习到的修复模式来增强或修正图像。
|
|
|
|
# Input types
|
|
## Required
|
|
- head
|
|
- head参数代表要加载的修复模型头部。它在决定修复行为和输出图像质量方面起着至关重要的作用。
|
|
- Comfy dtype: COMBO[STRING]
|
|
- Python dtype: str
|
|
- patch
|
|
- patch参数指的是要应用的特定修复补丁。这个补丁会修改模型的行为，允许基于提供的补丁信息进行有针对性的修复。
|
|
- Comfy dtype: COMBO[STRING]
|
|
- Python dtype: str
|
|
|
|
# Output types
|
|
- patch
|
|
- 输出是一个包含修复头部模型和LoRA补丁的元组，可以直接应用于模型以执行修复任务。
|
|
- Comfy dtype: INPAINT_PATCH
|
|
- Python dtype: Tuple[torch.nn.Module, dict]
|
|
|
|
|
|
## Usage tips
|
|
- Infra type: `CPU`
|
|
- Common nodes: unknown
|
|
|
|
|
|
## Source code
|
|
```python
|
|
class fooocusInpaintLoader:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"head": (list(FOOOCUS_INPAINT_HEAD.keys()),),
|
|
"patch": (list(FOOOCUS_INPAINT_PATCH.keys()),),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("INPAINT_PATCH",)
|
|
RETURN_NAMES = ("patch",)
|
|
CATEGORY = "EasyUse/Inpaint"
|
|
FUNCTION = "apply"
|
|
|
|
def apply(self, head, patch):
|
|
global inpaint_head_model
|
|
|
|
head_file = get_local_filepath(FOOOCUS_INPAINT_HEAD[head]["model_url"], INPAINT_DIR)
|
|
if inpaint_head_model is None:
|
|
inpaint_head_model = InpaintHead()
|
|
sd = torch.load(head_file, map_location='cpu')
|
|
inpaint_head_model.load_state_dict(sd)
|
|
|
|
patch_file = get_local_filepath(FOOOCUS_INPAINT_PATCH[patch]["model_url"], INPAINT_DIR)
|
|
inpaint_lora = comfy.utils.load_torch_file(patch_file, safe_load=True)
|
|
|
|
return ((inpaint_head_model, inpaint_lora),)
|
|
|
|
```
|