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