https://github.com/wsh032/kohya-config-webui
A WebUI for making config files used by kohya_sd_script. 一个用于生成kohya-ss训练脚本使用的toml配置文件的WebUI
https://github.com/wsh032/kohya-config-webui
colab kohya-sd-script stable-diffusion toml webui
Last synced: about 2 months ago
JSON representation
A WebUI for making config files used by kohya_sd_script. 一个用于生成kohya-ss训练脚本使用的toml配置文件的WebUI
- Host: GitHub
- URL: https://github.com/wsh032/kohya-config-webui
- Owner: WSH032
- License: apache-2.0
- Created: 2023-04-14T14:13:06.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2023-08-01T22:41:45.000Z (almost 2 years ago)
- Last Synced: 2025-03-23T01:01:39.042Z (2 months ago)
- Topics: colab, kohya-sd-script, stable-diffusion, toml, webui
- Language: Jupyter Notebook
- Homepage:
- Size: 80.1 KB
- Stars: 96
- Watchers: 1
- Forks: 4
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 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](https://github.com/WSH032/lora-scripts/) | 基于[Akegarasu/lora-scripts](https://github.com/Akegarasu/lora-scripts)的定制化Colab notebook | [](https://colab.research.google.com/github/WSH032/lora-scripts/blob/main/Colab_Lora_train.ipynb) | [](https://colab.research.google.com/drive/1_f0qJdM43BSssNJWtgjIlk9DkIzLPadx) |
| [kohya_train_webui](https://github.com/WSH032/kohya-config-webui) `NEW` | 基于[WSH032/kohya-config-webui](https://github.com/WSH032/kohya-config-webui)的WebUI版Colab notebook | [](https://colab.research.google.com/github/WSH032/kohya-config-webui/blob/main/kohya_train_webui.ipynb) |
## 使用方法
### （一）Colab版本（带有完整训练环境）：
[](https://colab.research.google.com/github/WSH032/kohya-config-webui/blob/main/kohya_train_webui.ipynb)
### （二）AUTOMATIC1111/stable-diffusion-webui插件：
`https://github.com/WSH032/kohya-config-webui.git`
将这个仓库连接复制到SD-WebUi的`扩展`->`从网址安装`界面下载即可
### （三）直接下载：
运行以下代码，或直接从github上下载zip并解压（直接下载将无法使用[update.bat](update.bat)），
```Shell
git clone https://github.com/WSH032/kohya-config-webui.git
```
运行目录下的[install_webui.ps1](install_webui.ps1)在虚拟环境中安装gradio和toml,或者
```Python
pip install gradio>=3.24.1
pip install toml>=0.10.2
```
运行[run_webui.ps1](run_webui.ps1)，或者
```Python
python .\module\kohya_config_webui.py
```
# Todo
- [x] 增加上一次参数保存功能
- [ ] 增加dataset_config.toml生成功能
# Credit
**Based on the work of [kohya-ss](https://github.com/kohya-ss/sd-scripts) , [Linaqruf](https://github.com/Linaqruf/kohya-trainer). Thanks to them.**
- 这个项目使用的的训练脚本来自[kohya-ss](https://github.com/kohya-ss/sd-scripts)
- notebook中部分代码（如下载模块）来自[Linaqruf](https://github.com/Linaqruf/kohya-trainer)
**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
如果你基于此项目进行了修改、引用等用途，请注意原作者的协议。
请在你使用的部分标明代码来源。
# 其他部分展示
