SD-Trainer

✨ 享受Stable Diffusion训练！ ✨

</div> <p align="center"> <a href="https://github.com/Akegarasu/lora-scripts" style="margin: 2px;"> <img alt="GitHub 仓库星标数" src="https://img.shields.io/github/stars/Akegarasu/lora-scripts"> </a> <a href="https://github.com/Akegarasu/lora-scripts" style="margin: 2px;"> <img alt="GitHub 分支数" src="https://img.shields.io/github/forks/Akegarasu/lora-scripts"> </a> <a href="https://raw.githubusercontent.com/Akegarasu/lora-scripts/master/LICENSE" style="margin: 2px;"> <img src="https://img.shields.io/github/license/Akegarasu/lora-scripts" alt="许可证"> </a> <a href="https://github.com/Akegarasu/lora-scripts/releases" style="margin: 2px;"> <img src="https://img.shields.io/github/v/release/Akegarasu/lora-scripts?color=blueviolet&include_prereleases" alt="发布版本"> </a> </p> <p align="center"> <a href="https://github.com/Akegarasu/lora-scripts/releases">下载</a> · <a href="https://github.com/Akegarasu/lora-scripts/blob/main/README.md">文档</a> · <a href="https://github.com/Akegarasu/lora-scripts/blob/main/README-zh.md">中文README</a> </p>

LoRA-scripts（又名SD-Trainer）

LoRA & Dreambooth训练GUI、脚本预设和一键训练环境，适用于kohya-ss/sd-scripts

✨新功能：训练WebUI

真正的Stable Diffusion训练工作室。一个WebUI集成所有功能。

按照下面的安装指南安装GUI，然后运行run_gui.ps1（Windows）或run_gui.sh（Linux）启动GUI。

Tensorboard	WD 1.4 标签器	标签编辑器

使用方法

必需依赖

Python 3.10和Git

克隆仓库及子模块

git clone --recurse-submodules https://github.com/Akegarasu/lora-scripts

✨ SD-Trainer GUI

Windows

安装

运行install.ps1将自动为您创建虚拟环境并安装必要的依赖。如果您在中国大陆，请使用install-cn.ps1

训练

运行run_gui.ps1，程序将自动打开http://127.0.0.1:28000

Linux

安装

运行install.bash将创建虚拟环境并安装必要的依赖。

训练

运行bash run_gui.bash，程序将自动打开http://127.0.0.1:28000

传统手动运行脚本训练

Windows

安装

运行install.ps1将自动为您创建虚拟环境并安装必要的依赖。

训练

编辑train.ps1，然后运行。

Linux

安装

运行install.bash将创建虚拟环境并安装必要的依赖。

训练

训练脚本train.sh不会为您激活虚拟环境。您应该先激活虚拟环境。

source venv/bin/activate

编辑train.sh，然后运行。

TensorBoard

运行tensorboard.ps1将在http://localhost:6006/启动TensorBoard

程序参数

参数名	类型	默认值	描述
`--host`	str	"127.0.0.1"	服务器主机名
`--port`	int	28000	运行服务器的端口
`--listen`	bool	false	启用服务器的监听模式
`--skip-prepare-environment`	bool	false	跳过环境准备步骤
`--disable-tensorboard`	bool	false	禁用TensorBoard
`--disable-tageditor`	bool	false	禁用标签编辑器
`--tensorboard-host`	str	"127.0.0.1"	运行TensorBoard的主机
`--tensorboard-port`	int	6006	运行TensorBoard的端口
`--localization`	str		界面的本地化设置
`--dev`	bool	false	开发者模式，禁用某些检查