笔记:Stable Diffusion+Cleaner改批量处理尝试
都是我养不起的小猫咪
2024年08月23日 15:41

Stable Diffusion+Cleaner环境搭建

Stable Diffusion最全保姆级安装教程(建议收藏)_stable diffusion安装-CSDN博客

安装git(,系统变量添加)

 

 

安装cuda(查找官网适配,系统变量添加)

 

 

安装Anconda(所有用户,忽视警告系统变量自动添加/或者手动添加,重启)

 

 

安装pycharm

 

Anconda搭建虚拟环境

 

Conda create -n (你的环境名,eg:stableDiff) python==3.10.6

【下载设定华清源或其他(网上随便搜),快,网址下搜不到新包时可重置恢复默认源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

pip config set global.extra-index-url "<url1> <url2>..."

重置:pip config unset global.index-url

 

在自己选定的文件夹下新建项目,目录下cmd打开terminal控制台下“git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git”或者自己在stable diffusion

 

打开pycharm,加载git完成的项目

右下角interpreter setting,选取自己创建的虚拟环境stableDiff下的python.exe

 

webui-user.bat里set COMMANDLINE_ARGS=--enable-insecure-extension-access(根据自己电脑配置选择)

Pycharm下terminal控制台执行“webui-user.bat”,会自动在项目下搭建venv

一般而言只要是python是3.10.6,就可以自动下载。外网下载缓慢,中断会从头下。

不行就停下,自己手动pip install** -i https://pypi.tuna.tsinghua.edu.cn/simple;

再不行,可网上搜包名,下载,中断可续下载

(主要是torch,我装的torch-2.1.2+cu121-cp310-cp310-win_amd64.whl,不能用xformers 加速,xformers现在最高支持torch2.1.1,可自行尝试配置调试)

 

下载cleaner插件

git clone https://github.com/novitalabs/sd-webui-cleaner.git

放入extensions文件夹内

 

big-lama.safetensors模型下载,

放入项目下extensions\sd-webui-cleaner\models

 

重新启动webui-user.bat,进入cleaner界面,即可进行手动消除。

Stable Diffusion+Cleaner编辑修改

进入项目下extensions\sd-webui-cleaner\scripts,

lama.py的clean_object(image,mask)函数中可添加修改mask的函数。

编写批量处理的代码,调用clean_object(image,mask),即可处理批量文件。

 

Stable Diffusion+Cleaner批量去水印打包

项目下Venv不知道为什么没法打包。

 

激活虚拟环境activate stableDiff

安装需要的包pip install “项目下的requirements.txt”,注意torch安装分cpu和gpu版本,根据自己需求安装。Gpu的慢

 

Pip install pyinstaller

Pyinstaller -D 自己写的py,例如removemarkers.txt(依赖包在exe外,出错好调整)

 

去dist下验证打包是否成功,

 

关于打包后模型的放置

在打包环境dist\_interal\scripts\lama.py里

(若没有,自己放)里LiteLama2(LiteLama)

添加MODEL_PATH = os.path.dirname(os.path.abspath(__file__)),

将模型放入与lama.py同级目录下,即打包环境dist\_interal\scripts下。

 

若缺包,去虚拟环境下找,或者用pycharm打开相应出错界面,control+鼠标点缺失的包。若存在,右键打开文件地址,将其复制到_interal下……循环往复,直至exe可执行。