
ComfyUI-SolAttn_triton
kijai
暂无描述
AI 简介
这是一个为ComfyUI集成Sol-Attn稀疏注意力机制的实验性扩展,基于Triton实现,用于加速扩散模型在图像与视频生成任务中的推理过程。核心功能是在不修改模型训练的前提下,通过动态稀疏化自注意力计算降低显存占用与延迟;技术特点包括Triton内核编译优化、可调参数(start_percent/end_percent/tau)控制稀疏强度与质量平衡,并已在RTX 4090/5090等消费级GPU上验证可用。适用于对生成速度和显存效率有要求的本地AI绘画、视频生成工作流。
Python
225
Stars
14
Forks
1
Watchers
2
Issues
Star 增长
今日0
近 7 天0
近 30 天0
综合评分37.53
默认分支main