Skip to content

Instantly share code, notes, and snippets.

@Birch-san
Last active September 11, 2022 23:19
Show Gist options
  • Save Birch-san/9518b55dc408e71c56b0c822b4ae65df to your computer and use it in GitHub Desktop.
Save Birch-san/9518b55dc408e71c56b0c822b4ae65df to your computer and use it in GitHub Desktop.
Training waifu-diffusion

Instructions for training waifu-diffusion, by dep from 東方Project AI Discord
https://canary.discord.com/channels/930499730843250783/950914873800396841/1018280003416367176

image

git clone https://github.com/harubaru/waifu-diffusion.git
cd waifu-diffusion 
pip install omegaconf einops pytorch-lightning==1.6.5 test-tube transformers kornia 
pip install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers 
pip install -e git+https://github.com/openai/CLIP.git@main#egg=clip
pip install setuptools==59.5.0
pip install pillow==9.0.1
pip install torchmetrics==0.6.0
pip install -e .

Download your dataset.zip
unzip dataset.zip
cd dataset
mkdir txt
mkdir img
mv .txt txt
mv img
cd ..
mv dataset danbooru-aesthetic

cp train.bat train.sh
nano train.sh
Add a coma next to the GPU count
Ctrl-X, save, yes, same name

chmod +x train.sh
./train.sh
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment