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。
适用于希望探索更快注意力实现的高阶用户。 |