CheckpointLoaderKJ

节点功能:用于加载和优化Stable Diffusion模型。

输出参数

参数名称说明
MODEL加载并应用补丁的模型结构,可用于后续推理(如 KSampler、生成流程等)。
CLIP模型中的 CLIP 模块,处理文本提示词输入。
VAE模型中的 VAE 解码器,用于将 latent 映射为图像。

控件参数

参数名称说明
ckpt_name要加载的模型检查点名称,从 ComfyUI/models/checkpoints 文件夹中选择。
patch_cublaslinear是否启用 CublasLinear 补丁,注意对已加载模型无效。勾选后将 torch.nn.Linear 替换为 CublasLinear,可提升推理速度(尤其在部分 GPU 架构上)。仅对新加载模型有效,建议首次加载前启用。
sage_attention是否启用 SageAttention 替代默认注意力机制,支持多种实现后端。 disabled:禁用替换,使用原生 attention。 auto:自动选择最优实现,推荐大多数用户使用。 sageattn_qk_int8_pv_fp16_cuda:CUDA后端,QK为int8,PV为fp16。 sageattn_qk_int8_pv_fp16_triton:Triton后端版本。 sageattn_qk_int8_pv_fp8_cuda:CUDA后端,QK为int8,PV为fp8。 适用于希望探索更快注意力实现的高阶用户。