闲谈【Stable-Diffusion WEBUI】的插件:模型工具箱:省空间利器

文章介绍了WEBUI的新插件ModelToolbox,主要用于模型的管理、编辑和创建。该工具能帮助用户修剪模型,减少模型大小,尤其是将模型转换为FP16精度以节省存储空间。用户还可以通过工具查看和替换模型组件,例如添加VAE编码器或解码器。此外,ModelToolbox具有自动修剪功能,可处理Autoprune目录下的模型并转换为FP16格式。
摘要由CSDN通过智能技术生成

(零)前言

本篇主要提到WEBUI的一个新插件,模型工具箱,可以修剪模型,提取修改模型组件信息。
更多不断丰富的内容参考:🔗《继续Stable-Diffusion WEBUI方方面面研究(内容索引)》

(一)模型工具箱(Model Toolbox)

参考:https://github.com/arenasys/stable-diffusion-webui-model-toolkit
可以从WEB UI中直接安装,它是一个用于管理、编辑和创建模型的多用途工具包。
安装后会出现新的选项页,主要功能有:

  • 清晰/修剪模型。
  • 转换到.safetensors格式。
  • 提取/替换模型组件。
  • 识别/调试模型架构。

由于许多我们得到的模型都非常臃肿,这些模型的大部分大小都是冗余或无用的数据。
所以作者开发了这个插件,简单的说可以减少模型大小(特别是FP16精度)。
当然同时可以查看/替换模型组件,所以具体它可以干这些事儿:

  1. .ckpt模型转换为.safetensors格式。
  2. 通过只保留FP16精度大幅缩小模型,省空间。
  3. VAE编解码器加入到没有包含VAE的模型。
  4. 查看模型信息,虽然不懂,但是好好玩。

(1.1)基本使用界面

首先是选择一个模型来加载:下拉列表(包括基础/LORA/VAE等模型)选中后点击【载入】按钮。
稍等片刻,模型加载完成后,可以看到模型的一些情况:
在这里插入图片描述
基本信息包括:

  • 模型的大小(字节)
  • 模型的类型(SD1.0 or SD2.0)
  • 模型包含的组件
  • 模型有没有垃圾数据
  • 当前选择精度下(主要是FP16)模型浪费了多少大小
  • 模型是否包含错误数据
  • 模型可以裁剪(缩小)到多大(字节)

点击【保存】按钮:可以将模型保存成当前选择精度的.safetensors格式文件。
文件名称会自动变成模型文件原名-xxxx-yyyy-zzzz.safetensors格式,后面三项分别对应UNET - VAE - CLIP

点击【清除】按钮:退出本次操作,回到初始的模型选择界面。

(1.2)高阶使用界面

就是可以导入(替换)或导出模型的组件。
参考官网仓库说明,比如我想导入(替换)一个VAE解码器到一个已有的模型。
只需要加载模型后,选高阶页面,选好架构(没得选)->类别(选VAE)->自动

然后下面文件选择一个已有的vae文件,点击【导入】,最后【保存】成新的模型文件就好了。
在这里插入图片描述
下面截图未能显示部分,还有更详细的统计信息。

(1.3)自动修剪模型

它还有个选项,打开后可以自动将models/Autoprune里面放置的模型(当然是我们自己放进去的)修剪到FP16精度,并自动移动到对应的文件夹(基础模型/LoRA模型/……)。

考虑到FP16精度其实很多时候和FP32差不太多。
且默认情况下,WEBUI会将所有加载的模型转换为 FP16(除非打开--no-half选项)。

这个插件应该很有用


🤪 happy saving space…

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值