mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 22:35:09 +00:00
100 lines
3.2 KiB
Markdown
100 lines
3.2 KiB
Markdown
This repository contains training, generation and utility scripts for Stable Diffusion.
|
|
|
|
[日本語版README](./README-ja.md)
|
|
|
|
For easier use (GUI and PowerShell scripts etc...), please visit [the repository maintained by bmaltais](https://github.com/bmaltais/kohya_ss). Thanks to @bmaltais!
|
|
|
|
This repository contains the scripts for:
|
|
|
|
* DreamBooth training, including U-Net and Text Encoder
|
|
* fine-tuning (native training), including U-Net and Text Encoder
|
|
* image generation
|
|
* model conversion (supports 1.x and 2.x, Stable Diffision ckpt/safetensors and Diffusers)
|
|
|
|
## About requirements_*.txt
|
|
|
|
These files do not contain requirements for PyTorch. Because the versions of them depend on your environment. Please install PyTorch at first (see installation guide below.)
|
|
|
|
The scripts are tested with PyTorch 1.12.1 and 1.13.0, Diffusers 0.10.2.
|
|
|
|
## Links to how-to-use documents
|
|
|
|
All documents are in Japanese currently, and CUI based.
|
|
|
|
* [Environment setup and DreamBooth training guide](https://note.com/kohya_ss/n/nee3ed1649fb6)
|
|
* [Fine-tuning step-by-step guide](https://note.com/kohya_ss/n/nbf7ce8d80f29):
|
|
Including BLIP captioning and tagging by DeepDanbooru or WD14 tagger
|
|
* [Image generation](https://note.com/kohya_ss/n/n2693183a798e)
|
|
* [Model conversion](https://note.com/kohya_ss/n/n374f316fe4ad)
|
|
|
|
## Windows Required Dependencies
|
|
|
|
Python 3.10.6 and Git:
|
|
|
|
- Python 3.10.6: https://www.python.org/ftp/python/3.10.6/python-3.10.6-amd64.exe
|
|
- git: https://git-scm.com/download/win
|
|
|
|
Give unrestricted script access to powershell so venv can work:
|
|
|
|
- Open an administrator powershell window
|
|
- Type `Set-ExecutionPolicy Unrestricted` and answer A
|
|
- Close admin powershell window
|
|
|
|
## Windows Installation
|
|
|
|
Open a regular Powershell terminal and type the following inside:
|
|
|
|
```powershell
|
|
git clone https://github.com/kohya-ss/sd-scripts.git
|
|
cd sd-scripts
|
|
|
|
python -m venv --system-site-packages venv
|
|
.\venv\Scripts\activate
|
|
|
|
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
|
|
pip install --upgrade -r requirements.txt
|
|
pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
|
|
|
|
cp .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\
|
|
cp .\bitsandbytes_windows\cextension.py .\venv\Lib\site-packages\bitsandbytes\cextension.py
|
|
cp .\bitsandbytes_windows\main.py .\venv\Lib\site-packages\bitsandbytes\cuda_setup\main.py
|
|
|
|
accelerate config
|
|
|
|
```
|
|
|
|
Answers to accelerate config:
|
|
|
|
```txt
|
|
- 0
|
|
- 0
|
|
- NO
|
|
- NO
|
|
- All
|
|
- fp16
|
|
```
|
|
|
|
## Upgrade
|
|
|
|
When a new release comes out you can upgrade your repo with the following command:
|
|
|
|
```powershell
|
|
cd sd-scripts
|
|
git pull
|
|
.\venv\Scripts\activate
|
|
pip install --upgrade -r requirements.txt
|
|
```
|
|
|
|
Once the commands have completed successfully you should be ready to use the new version.
|
|
|
|
## License
|
|
|
|
The majority of scripts is licensed under ASL 2.0 (including codes from Diffusers), however portions of the project are available under separate license terms:
|
|
|
|
[Memory Efficient Attention Pytorch](https://github.com/lucidrains/memory-efficient-attention-pytorch): MIT
|
|
|
|
[bitsandbytes](https://github.com/TimDettmers/bitsandbytes): MIT
|
|
|
|
[BLIP](https://github.com/salesforce/BLIP): BSD-3-Clause
|
|
|