Home

Awesome

Waifu Diffusion

Waifu Diffusion is the name for this project of finetuning Stable Diffusion on anime-styled images.

<img src=https://user-images.githubusercontent.com/26317155/194690196-8da73f2a-039d-4349-8b08-e24e8fd20959.png width=40% height=40%>

<sub>1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt</sub>

Setup

pip install -r requirements.txt

Project Structure

├── dataset: Dataset preparation and utilities
│   ├── aesthetic: Aesthetic ranking
│   └── download: Downloading utilities
└── trainer: The actual training code

License

Training Code: AGPL-3.0 Model Weights: CreativeML Open RAIL-M

Discord Server