├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── consistency_models ├── __init__.py ├── consistency_models.py └── utils.py ├── notebooks ├── consistency_models_training_example.ipynb └── improved_techniques_for_consistency_training_example.ipynb ├── requirements.txt ├── setup.cfg └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/README.md -------------------------------------------------------------------------------- /consistency_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/consistency_models/__init__.py -------------------------------------------------------------------------------- /consistency_models/consistency_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/consistency_models/consistency_models.py -------------------------------------------------------------------------------- /consistency_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/consistency_models/utils.py -------------------------------------------------------------------------------- /notebooks/consistency_models_training_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/notebooks/consistency_models_training_example.ipynb -------------------------------------------------------------------------------- /notebooks/improved_techniques_for_consistency_training_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/notebooks/improved_techniques_for_consistency_training_example.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch==1.13.1 2 | tqdm==4.64.0 3 | -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [isort] 2 | profile = black -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kinyugo/consistency_models/HEAD/setup.py --------------------------------------------------------------------------------