consistency_models
Consistency Model Framework
A PyTorch-based framework for training and sampling consistency models in image generation
Official repo for consistency models.
6k stars
59 watching
425 forks
Language: Python
last commit: over 1 year ago Related projects:
| Repository | Description | Stars |
|---|---|---|
| openai/guided-diffusion | This project is a software implementation of a diffusion model architecture, allowing users to generate synthetic images based on a learned distribution. | 6,366 |
| openai/baselines | High-quality implementations of reinforcement learning algorithms for research and development purposes | 15,885 |
| openai/gpt-2 | A repository providing code and models for research into language modeling and multitask learning | 22,644 |
 lucidrains/dalle2-pytorch | An implementation of DALL-E 2's text-to-image synthesis neural network in PyTorch | 11,184 |
 quark0/darts | Automated method to find optimal architecture for machine learning models | 3,926 |
 stability-ai/stablelm | Develops and maintains large language models with improved stability and performance | 15,829 |
 antixk/pytorch-vae | A collection of Variational AutoEncoder implementations in PyTorch | 6,776 |
 cadene/pretrained-models.pytorch | Provides pre-trained PyTorch models for various computer vision tasks and datasets. | 9,044 |
 lukemelas/efficientnet-pytorch | A PyTorch implementation of EfficientNet convolutional neural networks | 7,971 |
 postech-cvlab/pytorch-studiogan | A comprehensive PyTorch library providing implementations of various Generative Adversarial Networks (GANs) for image generation and evaluation. | 3,439 |
 clovaai/stargan-v2 | A Python implementation of an image-to-image translation model for generating diverse images across multiple domains. | 3,513 |
| stability-ai/stablediffusion | A software project that enables high-resolution image synthesis through a specific type of generative model using latent diffusion processes. | 39,501 |
 cszn/kair | Image restoration toolbox with training and testing codes for various deep learning-based methods | 2,994 |
 cornellius-gp/gpytorch | A library for creating scalable and flexible Gaussian process models with ease | 3,605 |
 lllyasviel/controlnet | An implementation of a neural network structure to control diffusion models by adding extra conditions. | 30,944 |