AnimeFlow / README.md
ntt123's picture
Upload folder using huggingface_hub
0c9bb32 verified
|
raw
history blame
508 Bytes
metadata
title: AnimeFlow
app_file: app.py
sdk: gradio
sdk_version: 5.9.1

Anime Flow

A simple implementation of conditional flow matching for generating anime faces. The model architecture closely follows the Diffusion Transformer model (DiT) found at https://github.com/facebookresearch/DiT/blob/main/models.py.

Train model

pip install uv
uv run train.py --config ./config.yaml

Generate images

uv run sample.py --ckpt ./state_1000000.ckpt --config ./config.yaml --seed 0