harubaru / waifu-diffusion

stable diffusion finetuned on weeb stuff

Waifu Diffusion

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

_{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