consistency_models

该项目是OpenAI官方发布的用于实现一致性模型的代码库，主要基于PyTorch框架。它支持在ImageNet-64、LSUN Bedroom-256和LSUN Cat-256等数据集上进行大规模实验，包括一致性蒸馏、一致性训练以及多种采样和编辑算法。项目提供了预训练模型供下载，并详细说明了每个模型的使用场景与限制条件。适用于需要高质量图像生成或处理的研究者及开发者，在图像生成、编辑等领域具有广泛的应用潜力。