最强（之一）图像分割节点ComfyUI SDMatte修改版，模型减小至2g，显存占用降低50%，8g显卡可用。
简单来说，原始模型权重精度是32位，内部还包括6g多的训练用权重和实际用不到的文本编码器权重，因此我将不需要的部分剥离，并将模型保存为16位精度。所以现在模型不到2g，显存占用大概6g。使用时记得在节点上切换对应精度。模型可以自行下载也可以选择一个运行节点，会自动下载（从huggingface，所以对网络有要求），选择pruned版即可。我只是基于前人的工作做了些小修改，大伙儿记得感谢原作者@Smiles_wx，以及为原项目点个星。链接如下：
https://github.com/flybirdxx/Co
立即观看
0