Waifu Diffusion

November 10, 2022 · View on GitHub

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%>

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

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