---
title: AnimeFlow
app_file: app.py
sdk: gradio
sdk_version: 5.9.1
short_description: Generate anime faces with conditional flow matching
---
# 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

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

## Generate images

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