Loading…
NVlabs/TCM is the official PyTorch implementation of Truncated Consistency Models (TCM), a generative model introduced at ICLR 2025. It provides training and evaluation pipelines, pre-trained checkpoints for TCM, EDM, and EDM2, and experiments on CIFAR-10 and ImageNet (64x64) using DDPM++-style variants. The repository includes tools and scripts (ct_train.py, ct_eval.py, dataset_tool.py, generate.py) and environment specifications, designed to run on NVIDIA A100 GPUs.
AI named TCM (Truncated Consistency Models) in February 2026.