mlfoundations / open-diffusion
Simple large-scale training of stable diffusion with multi-node support.
β128Updated last year
Alternatives and similar repositories for open-diffusion:
Users that are interested in open-diffusion are comparing it to the libraries listed below
- Iterable datapipelines for pytorch training.β81Updated 5 months ago
- Diffusion Reinforcement Learning Libraryβ179Updated last year
- π€ Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorchβ50Updated 2 years ago
- β83Updated last year
- An in-context conditioning version of MUSE with pre-trained checkpoints.β111Updated last year
- JAX implementation ViT-VQGANβ80Updated 2 years ago
- CLOOB training (JAX) and inference (JAX and PyTorch)β70Updated 2 years ago
- [ICLR 2025] Official PyTorch implmentation of paper "T-Stitch: Accelerating Sampling in Pre-trained Diffusion Models with Trajectory Stitβ¦β99Updated 11 months ago
- β164Updated last year
- A Video Tokenizer Evaluation Datasetβ101Updated last month
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"β167Updated 8 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more sampβ¦β271Updated 3 months ago
- π¦Ύ EvalGIM (pronounced as "EvalGym") is an evaluation library for generative image models. It enables easy-to-use, reproducible automaticβ¦β66Updated 2 months ago
- β72Updated last year
- Let's make a video clipβ93Updated 2 years ago
- MoVQGAN - model for the image encoding and reconstructionβ218Updated last year
- Open reproduction of MUSE for fast text2image generation.β342Updated 8 months ago
- Educational repository for applying the main video data curation techniques presented in the Stable Video Diffusion paper.β81Updated last year
- Official implementation of "Controlling Text-to-Image Diffusion by Orthogonal Finetuning".β285Updated 3 months ago
- Implementation of the video diffusion model and training scheme presented in the paper, Flexible Diffusion Modeling of Long Videos, in Pyβ¦β84Updated 2 years ago
- β52Updated last year
- Reproduction of DDPO paper (RLHF for diffusion)β81Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorchβ305Updated last month
- Faster generation with text-to-image diffusion models.β210Updated 4 months ago
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlightβ129Updated last year
- Huggingface-compatible SDXL Unet implementation that is readily hackableβ410Updated last year
- β102Updated last year
- Train VAE like a bossβ265Updated 3 months ago
- SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Trainingβ174Updated 3 weeks ago
- Code for instruction-tuning Stable Diffusion.β221Updated last year