kohya-config-webui

A WebUI for making config files used by kohya_sd_script

现在我们有什么？

目前，我为kohya-lora-dreambooth训练中较为常用、实用的训练参数，使用gradio和toml库，编写了一个交互式的WebUI生成工具，可以在带有python环境的windows和Colab环境中快速部署。

使用本项目，你可以快速指定训练参数，并生成config_file.toml和sample_prompt.txt。

如果你觉得此项目有用，可以给我一颗小星星，非常感谢你⭐

---

Notebook Name	Description	Link	Old-Version
Colab_Lora_train	基于Akegarasu/lora-scripts的定制化Colab notebook
kohya_train_webui NEW	基于WSH032/kohya-config-webui的WebUI版Colab notebook

使用方法

（一）Colab版本（带有完整训练环境）：

（二）AUTOMATIC1111/stable-diffusion-webui插件：

https://github.com/WSH032/kohya-config-webui.git

将这个仓库连接复制到SD-WebUi的扩展->从网址安装界面下载即可

（三）直接下载：

运行以下代码，或直接从github上下载zip并解压（直接下载将无法使用update.bat），

git clone https://github.com/WSH032/kohya-config-webui.git

运行目录下的install_webui.ps1在虚拟环境中安装gradio和toml,或者

pip install  gradio>=3.24.1
pip install  toml>=0.10.2

运行run_webui.ps1，或者

python .\module\kohya_config_webui.py

Todo

• 增加上一次参数保存功能
• 增加dataset_config.toml生成功能

Credit

Based on the work of kohya-ss , Linaqruf. Thanks to them.

• 这个项目使用的的训练脚本来自kohya-ss
• notebook中部分代码（如下载模块）来自Linaqruf

Attention: It's called kohya-config-webui, but I don't have a license for kohya. It just creates config files for kohya-ss.

• 这个插件叫kohya-config-webui，但是我没有kohya的授权，我只是为了说明它的作用是生成一个用于kohya-ss训练的config文件

---

上述两位作者和我目前采取的是Apache-2.0 license

如果你基于此项目进行了修改、引用等用途，请注意原作者的协议。

请在你使用的部分标明代码来源。

其他部分展示