Shypd.aiThe largest open index of AI toolsImproved-diffusion is an open-source codebase for Denoising Diffusion Probabilistic Models, enabling users to train and sample from these advanced AI models. It provides scripts for installation, data preparation, and model usage, supporting both unconditional and class-conditional image generation.
Trending
Also listed in
About
Improved-diffusion is an open-source codebase developed by OpenAI for working with Improved Denoising Diffusion Probabilistic Models. This repository provides the necessary tools and scripts for researchers and developers to train and sample from these powerful generative AI models. Users can prepare their own image datasets, including options for class-conditional training by naming files with labels. The codebase supports various hyperparameters for model architecture, diffusion processes, and training flags, allowing for flexible experimentation. It also facilitates distributed training across multiple GPUs and offers different sampling strategies, including DDIM. Pre-trained model checkpoints and their corresponding hyperparameters are provided for several common tasks, such as unconditional ImageNet-64 and CIFAR-10 generation, class-conditional ImageNet-64, and LSUN bedroom models.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
