File size: 578 Bytes
225efdd
 
0c9bb32
225efdd
 
c96e9fc
225efdd
0c9bb32
 
 
 
 
 
 
 
 
 
 
 
225efdd
0c9bb32
 
c96e9fc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
---
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
```