CheckpointLoaderKJ
节点功能:用于加载和优化Stable Diffusion模型。
节点中英文对比
CheckpointLoaderKJ
ckpt_name
patch_cublaslinear
sage_attention
CheckpointLoaderKJ
ckpt_name
patch_cublaslinear
sage_attention
CheckpointLoaderKJ - 参数说明
输出参数
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。
适用于希望探索更快注意力实现的高阶用户。
暂无节点说明