Skip to content

LoRA & Dreambooth training scripts & GUI use kohya-ss's trainer, for diffusion model.

License

Notifications You must be signed in to change notification settings

weilin9999/lora-scripts

This branch is 160 commits behind Akegarasu/lora-scripts:main.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

1f65891 · Feb 7, 2024
Nov 15, 2023
Aug 31, 2023
Feb 6, 2024
Jan 22, 2023
Jan 23, 2023
Jan 27, 2024
Jan 22, 2023
Jan 22, 2023
Jan 16, 2024
Feb 10, 2023
Dec 8, 2023
Nov 15, 2023
Nov 16, 2023
Jun 1, 2023
Jan 15, 2024
Jan 15, 2024
Dec 29, 2023
Oct 16, 2023
Dec 20, 2023
Sep 11, 2023
Mar 1, 2023
Feb 7, 2024
Feb 7, 2024
Mar 26, 2023
Nov 16, 2023
Apr 27, 2023
May 20, 2023
May 14, 2023
Jul 31, 2023
Jul 31, 2023
Apr 27, 2023
Jan 26, 2023
Jan 21, 2024
Jan 21, 2024
Jan 31, 2024
Jan 31, 2024

Repository files navigation

SD-Trainer

SD-Trainer

✨ Enjoy Stable Diffusion Train! ✨

GitHub Repo stars GitHub forks license release

Download · Documents · 中文README

LoRA-scripts (a.k.a SD-Trainer)

LoRA & Dreambooth training GUI & scripts preset & one key training environment for kohya-ss/sd-scripts

✨NEW: Train WebUI

The REAL Stable Diffusion Training Studio. Everything in one WebUI.

Follow the installation guide below to install the GUI, then run run_gui.ps1(windows) or run_gui.sh(linux) to start the GUI.

image

Tensorboard WD 1.4 Tagger Tag Editor
image image image

Usage

Required Dependencies

Python 3.10 and Git

Clone repo with submodules

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

✨ SD-Trainer GUI

Windows

Installation

Run install.ps1 will automaticilly create a venv for you and install necessary deps. If you are in China mainland, please use install-cn.ps1

Train

run run_gui.ps1, then program will open http://127.0.0.1:28000 automanticlly

Linux

Installation

Run install.bash will create a venv and install necessary deps.

Train

run bash run_gui.bash, then program will open http://127.0.0.1:28000 automanticlly

Legacy training through run script manually

Windows

Installation

Run install.ps1 will automaticilly create a venv for you and install necessary deps.

Train

Edit train.ps1, and run it.

Linux

Installation

Run install.bash will create a venv and install necessary deps.

Train

Training script train.sh will not activate venv for you. You should activate venv first.

source venv/bin/activate

Edit train.sh, and run it.

TensorBoard

Run tensorboard.ps1 will start TensorBoard at http://localhost:6006/

Program arguments

Parameter Name Type Default Value Description
--host str "127.0.0.1" Hostname for the server
--port int 28000 Port to run the server
--listen bool false Enable listening mode for the server
--skip-prepare-environment bool false Skip the environment preparation step
--disable-tensorboard bool false Disable TensorBoard
--disable-tageditor bool false Disable tag editor
--tensorboard-host str "127.0.0.1" Host to run TensorBoard
--tensorboard-port int 6006 Port to run TensorBoard
--localization str Localization settings for the interface
--dev bool false Developer mode to disale some checks

About

LoRA & Dreambooth training scripts & GUI use kohya-ss's trainer, for diffusion model.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 56.5%
  • PowerShell 21.0%
  • Shell 15.6%
  • Jupyter Notebook 6.3%
  • Dockerfile 0.6%