├── .github └── workflows │ └── publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── assets ├── consistency_models.png └── training.gif ├── consistency ├── __init__.py ├── consistency.py ├── loss.py └── pipeline.py ├── examples ├── consistency_models.ipynb ├── fid_score │ ├── .gitignore │ ├── calculate_fid_score.py │ └── requirements.txt ├── sample.ipynb └── unconditional_image_generation │ ├── README.md │ ├── requirements.txt │ └── train_unconditional.py └── setup.py /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/README.md -------------------------------------------------------------------------------- /assets/consistency_models.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/assets/consistency_models.png -------------------------------------------------------------------------------- /assets/training.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/assets/training.gif -------------------------------------------------------------------------------- /consistency/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/consistency/__init__.py -------------------------------------------------------------------------------- /consistency/consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/consistency/consistency.py -------------------------------------------------------------------------------- /consistency/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/consistency/loss.py -------------------------------------------------------------------------------- /consistency/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/consistency/pipeline.py -------------------------------------------------------------------------------- /examples/consistency_models.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/examples/consistency_models.ipynb -------------------------------------------------------------------------------- /examples/fid_score/.gitignore: -------------------------------------------------------------------------------- 1 | evaluation -------------------------------------------------------------------------------- /examples/fid_score/calculate_fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/examples/fid_score/calculate_fid_score.py -------------------------------------------------------------------------------- /examples/fid_score/requirements.txt: -------------------------------------------------------------------------------- 1 | datasets 2 | diffusers 3 | clean-fid 4 | -------------------------------------------------------------------------------- /examples/sample.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/examples/sample.ipynb -------------------------------------------------------------------------------- /examples/unconditional_image_generation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/examples/unconditional_image_generation/README.md -------------------------------------------------------------------------------- /examples/unconditional_image_generation/requirements.txt: -------------------------------------------------------------------------------- 1 | huggingface-hub 2 | datasets 3 | wandb 4 | consistency 5 | -------------------------------------------------------------------------------- /examples/unconditional_image_generation/train_unconditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/examples/unconditional_image_generation/train_unconditional.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhsss/consistency-models/HEAD/setup.py --------------------------------------------------------------------------------