consistency_models

Consistency Model Framework

A PyTorch-based framework for training and sampling consistency models in image generation

Official repo for consistency models.

GitHub

6k stars
59 watching
425 forks
Language: Python
last commit: over 2 years ago

Related projects:

RepositoryDescriptionStars
openai/guided-diffusionThis project is a software implementation of a diffusion model architecture, allowing users to generate synthetic images based on a learned distribution.6,366
openai/baselinesHigh-quality implementations of reinforcement learning algorithms for research and development purposes15,885
openai/gpt-2A repository providing code and models for research into language modeling and multitask learning22,644
lucidrains/dalle2-pytorchAn implementation of DALL-E 2's text-to-image synthesis neural network in PyTorch11,184
quark0/dartsAutomated method to find optimal architecture for machine learning models3,926
stability-ai/stablelmDevelops and maintains large language models with improved stability and performance15,829
antixk/pytorch-vaeA collection of Variational AutoEncoder implementations in PyTorch6,776
cadene/pretrained-models.pytorchProvides pre-trained PyTorch models for various computer vision tasks and datasets.9,044
lukemelas/efficientnet-pytorchA PyTorch implementation of EfficientNet convolutional neural networks7,971
postech-cvlab/pytorch-studioganA comprehensive PyTorch library providing implementations of various Generative Adversarial Networks (GANs) for image generation and evaluation.3,439
clovaai/stargan-v2A Python implementation of an image-to-image translation model for generating diverse images across multiple domains.3,513
stability-ai/stablediffusionA software project that enables high-resolution image synthesis through a specific type of generative model using latent diffusion processes.39,501
cszn/kairImage restoration toolbox with training and testing codes for various deep learning-based methods2,994
cornellius-gp/gpytorchA library for creating scalable and flexible Gaussian process models with ease3,605
lllyasviel/controlnetAn implementation of a neural network structure to control diffusion models by adding extra conditions.30,944