├── .gitignore ├── LICENSE ├── README.md ├── assets ├── cfg_200k_cfg2.png ├── meanflow.gif ├── mnist_10k.png └── mnist_6k_cfg2.png ├── meanflow.py ├── models └── dit.py ├── train.py └── train_latent.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/README.md -------------------------------------------------------------------------------- /assets/cfg_200k_cfg2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/assets/cfg_200k_cfg2.png -------------------------------------------------------------------------------- /assets/meanflow.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/assets/meanflow.gif -------------------------------------------------------------------------------- /assets/mnist_10k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/assets/mnist_10k.png -------------------------------------------------------------------------------- /assets/mnist_6k_cfg2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/assets/mnist_6k_cfg2.png -------------------------------------------------------------------------------- /meanflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/meanflow.py -------------------------------------------------------------------------------- /models/dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/models/dit.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/train.py -------------------------------------------------------------------------------- /train_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haidog-yaqub/MeanFlow/HEAD/train_latent.py --------------------------------------------------------------------------------