Diffusion Reinforcement Learning Library
☆195Feb 13, 2024Updated 2 years ago
Alternatives and similar repositories for DRLX
Users that are interested in DRLX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Reproduction of DDPO paper (RLHF for diffusion)☆94Sep 20, 2023Updated 2 years ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆325Nov 1, 2024Updated last year
- Code for the paper "Training Diffusion Models with Reinforcement Learning"☆578Jul 5, 2023Updated 3 years ago
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆775Mar 22, 2024Updated 2 years ago
- ☆22Oct 20, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- This repository implements the idea of "caption upsampling" from DALL-E 3 with Zephyr-7B and gathers results with SDXL.☆157Oct 25, 2023Updated 2 years ago
- Source code for "A Dense Reward View on Aligning Text-to-Image Diffusion with Preference" (ICML'24).☆39May 9, 2024Updated 2 years ago
- [NeurIPS 2023] ImageReward: Learning and Evaluating Human Preferences for Text-to-image Generation☆1,702Oct 29, 2025Updated 9 months ago
- ☆58Sep 23, 2024Updated last year
- An attempt at a SVD inpainting pipeline☆50Dec 24, 2023Updated 2 years ago
- ☆21Aug 27, 2023Updated 2 years ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆707Jun 2, 2026Updated 2 months ago
- [NeurIPS 2024] ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimization☆166Sep 15, 2025Updated 10 months ago
- ☆600Dec 21, 2024Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- SEIKO is a novel reinforcement learning method to efficiently fine-tune diffusion models in an online setting. Our methods outperform all…☆30Jul 18, 2024Updated 2 years ago
- mask2former psg☆22Dec 12, 2022Updated 3 years ago
- ☆33Aug 9, 2024Updated 2 years ago
- ☆53Jan 6, 2024Updated 2 years ago
- [ICLR 2024 Spotlight] Official implementation of ScaleCrafter for higher-resolution visual generation at inference time.☆507Mar 7, 2024Updated 2 years ago
- Pipeline to scrape prompt + image url pairs from LAION `share-dalle-3` discord channel☆11Oct 10, 2023Updated 2 years ago
- Recaption large (Web)Datasets with vllm and save the artifacts.☆53Nov 23, 2024Updated last year
- SigLIP-based Aesthetic Score Predictor☆433Dec 18, 2024Updated last year
- The official implementation of Diffusion-KTO: Aligning Diffusion Models by Optimizing Human Utility☆68Aug 16, 2025Updated 11 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A simple aesthetic scorer + pruner + website you can run to view the results from the scoring with☆16Jun 3, 2024Updated 2 years ago
- "FreeU: Free Lunch in Diffusion U-Net" for Huggingface Diffusers☆103Oct 6, 2023Updated 2 years ago
- Just a script i use for training YOLOs with most of parameters exposed and described.☆13Dec 13, 2024Updated last year
- Train vision models using JAX and 🤗 transformers☆104Dec 14, 2025Updated 7 months ago
- Implementation of HyperDreamBooth: HyperNetworks for Fast Personalization of Text-to-Image Models☆174Jul 19, 2023Updated 3 years ago
- Tools i've created(likely all of them will be with GPT4 usage) for processing and possible enchancement of dataset tagging☆15Mar 14, 2024Updated 2 years ago
- FreeU: Free Lunch in Diffusion U-Net (CVPR2024 Oral)☆1,901Dec 24, 2024Updated last year
- Cool extensions built for the nbdev framework☆14Jun 6, 2023Updated 3 years ago
- Train high-quality text-to-image diffusion models in a data & compute efficient manner☆515Jun 25, 2026Updated last month
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,753Jan 8, 2024Updated 2 years ago
- Official implementation of Würstchen: Efficient Pretraining of Text-to-Image Models☆554Apr 6, 2024Updated 2 years ago
- ✨ Hotshot-XL: State-of-the-art AI text-to-GIF model trained to work alongside Stable Diffusion XL☆1,111Jan 23, 2024Updated 2 years ago
- Official implementation of CVPR 2024 paper: "FreeControl: Training-Free Spatial Control of Any Text-to-Image Diffusion Model with Any Con…☆479Oct 21, 2024Updated last year
- End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).☆400Jan 8, 2026Updated 7 months ago
- Consistency Distilled Diff VAE☆2,213Nov 7, 2023Updated 2 years ago
- Train a production grade GPT in less than 400 lines of code. Better than Karpathy's verison and GIGAGPT☆17Updated this week