advanced/model_merginggenerated
Model Save KJ(ModelSaveKJ)
从技术实现来看,此节点接收一个 `MODEL` 类型的输入,通过 `state_dict_for_saving` 方法获取模型的参数字典
Model Save KJ
model
filename_prefix
diffusion_models/ComfyUI
model_key_prefix
model.diffusion_model.
KJNodes
从技术实现来看,此节点接收一个 MODEL 类型的输入,通过 state_dict_for_saving 方法获取模型的参数字典。
Model Save KJ 是一个用于保存经过处理或合并后的扩散模型权重的节点。它的核心功能是将工作流中的模型以 .safetensors 格式持久化到磁盘,并允许用户自定义保存路径和模型内部的键名前缀。该节点不输出任何数据,仅执行保存操作。
节点功能
从技术实现来看,此节点接收一个 MODEL 类型的输入,通过 state_dict_for_saving 方法获取模型的参数字典。其核心机制在于,它允许用户指定一个 model_key_prefix,用于替换模型参数字典中默认的键名前缀(model.diffusion_model.),然后将处理后的参数字典保存为 .safetensors 文件。这为后续加载到其他支持不同键名约定的框架或工具中提供了灵活性。
节点参数说明 - Model Save KJ
连接输入 (Inputs)
| 参数名称 | 数据类型 | 必填 | 默认值 | 取值范围/选项 | 说明 |
|---|---|---|---|---|---|
model | MODEL | 是 | - | - | 指定需要保存的模型对象。这是节点操作的核心数据源。 |
控件参数 (Parameters)
| 参数名称 | 数据类型 | 必填 | 默认值 | 取值范围/选项 | 说明 |
|---|---|---|---|---|---|
filename_prefix | STRING | 是 | diffusion_models/ComfyUI | - | 定义保存文件的路径前缀和基础文件名。默认保存在 output 目录下的 diffusion_models/ComfyUI 路径中,文件名会自动添加序号。 |
model_key_prefix | STRING | 是 | model.diffusion_model. | - | 指定在保存的模型文件中,用于替换原始键名前缀的新字符串。默认值为 model.diffusion_model.,修改此值可以适配不同框架的加载要求。 |
使用场景
一个典型的使用场景是在模型合并或微调工作流之后。例如,当你使用多个 LoRA 模型对基础模型进行融合调整后,可以将最终生成的模型通过此节点保存下来。这样,你就能在其他项目或 ComfyUI 的其他工作流中直接加载这个自定义模型,而无需重复整个复杂的合并过程。
Model Save KJ 节点源码链接
Model Save KJ 节点来自 ComfyUI-KJNodes 节点包。
评论
使用 GitHub 登录后即可参与讨论。