83 lines
3.6 KiB
Markdown
83 lines
3.6 KiB
Markdown
# Documentation
|
|
- Class name: WAS_Lora_Loader
|
|
- Category: WAS Suite/Loaders
|
|
- Output node: False
|
|
- Repo Ref: https://github.com/WASasquatch/was-node-suite-comfyui
|
|
|
|
WAS_Lora_Loader节点旨在WAS套件中管理Lora模型的加载和应用。它通过缓存先前加载的模型来确保高效处理Lora文件，减少冗余并提高性能。此节点对于将高级定制功能集成到工作流程中至关重要。
|
|
|
|
# Input types
|
|
## Required
|
|
- model
|
|
- 模型参数对于节点的操作至关重要，因为它代表了将应用Lora增强的主要对象。它直接影响节点的执行和结果模型的能力。
|
|
- Comfy dtype: MODEL
|
|
- Python dtype: torch.nn.Module
|
|
- clip
|
|
- clip参数对于定义Lora模型操作的上下文至关重要。它影响模型如何与系统的其余部分集成，影响最终结果。
|
|
- Comfy dtype: CLIP
|
|
- Python dtype: Union[torch.Tensor, comfy.models.CLIPModel]
|
|
- lora_name
|
|
- lora_name参数指定要加载的Lora模型。它是定制过程中的关键决定因素，因为它决定了将使用哪个Lora模型的功能。
|
|
- Comfy dtype: STRING
|
|
- Python dtype: str
|
|
- strength_model
|
|
- strength_model参数调整Lora模型对基础模型影响的强度。它是微调模型性能的关键因素。
|
|
- Comfy dtype: FLOAT
|
|
- Python dtype: float
|
|
- strength_clip
|
|
- strength_clip参数控制CLIP模型对整个过程的影响。它在根据所需规格形成最终输出方面具有重要意义。
|
|
- Comfy dtype: FLOAT
|
|
- Python dtype: float
|
|
|
|
# Output types
|
|
- MODEL
|
|
- 输出模型代表了应用了Lora定制的增强模型。它标志着节点处理的完成，是工作流中后续步骤的核心。
|
|
- Comfy dtype: MODEL
|
|
- Python dtype: torch.nn.Module
|
|
- CLIP
|
|
- 输出CLIP模型反映了通过与Lora模型集成所做的调整。它是进一步处理或分析的重要组成部分。
|
|
- Comfy dtype: CLIP
|
|
- Python dtype: Union[torch.Tensor, comfy.models.CLIPModel]
|
|
- NAME_STRING
|
|
- NAME_STRING输出提供了加载的Lora模型的名称，这对于系统内的跟踪和引用非常有用。
|
|
- Comfy dtype: STRING
|
|
- Python dtype: str
|
|
|
|
# Usage tips
|
|
- Infra type: CPU
|
|
|
|
# Source code
|
|
```
|
|
class WAS_Lora_Loader:
|
|
|
|
def __init__(self):
|
|
self.loaded_lora = None
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
file_list = comfy_paths.get_filename_list('loras')
|
|
file_list.insert(0, 'None')
|
|
return {'required': {'model': ('MODEL',), 'clip': ('CLIP',), 'lora_name': (file_list,), 'strength_model': ('FLOAT', {'default': 1.0, 'min': -10.0, 'max': 10.0, 'step': 0.01}), 'strength_clip': ('FLOAT', {'default': 1.0, 'min': -10.0, 'max': 10.0, 'step': 0.01})}}
|
|
RETURN_TYPES = ('MODEL', 'CLIP', TEXT_TYPE)
|
|
RETURN_NAMES = ('MODEL', 'CLIP', 'NAME_STRING')
|
|
FUNCTION = 'load_lora'
|
|
CATEGORY = 'WAS Suite/Loaders'
|
|
|
|
def load_lora(self, model, clip, lora_name, strength_model, strength_clip):
|
|
if strength_model == 0 and strength_clip == 0:
|
|
return (model, clip)
|
|
lora_path = comfy_paths.get_full_path('loras', lora_name)
|
|
lora = None
|
|
if self.loaded_lora is not None:
|
|
if self.loaded_lora[0] == lora_path:
|
|
lora = self.loaded_lora[1]
|
|
else:
|
|
temp = self.loaded_lora
|
|
self.loaded_lora = None
|
|
del temp
|
|
if lora is None:
|
|
lora = comfy.utils.load_torch_file(lora_path, safe_load=True)
|
|
self.loaded_lora = (lora_path, lora)
|
|
(model_lora, clip_lora) = comfy.sd.load_lora_for_models(model, clip, lora, strength_model, strength_clip)
|
|
return (model_lora, clip_lora, os.path.splitext(os.path.basename(lora_name))[0])
|
|
``` |