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…☆324Nov 1, 2024Updated last year
- Code for the paper "Training Diffusion Models with Reinforcement Learning"☆573Jul 5, 2023Updated 3 years ago
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆768Mar 22, 2024Updated 2 years ago
- ☆22Oct 20, 2023Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- This repository implements the idea of "caption upsampling" from DALL-E 3 with Zephyr-7B and gathers results with SDXL.☆158Oct 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,694Oct 29, 2025Updated 8 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"☆705Jun 2, 2026Updated last month
- [NeurIPS 2024] ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimization☆166Sep 15, 2025Updated 10 months ago
- ☆601Dec 21, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- 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
- StableFused is a toy library for experimenting with Diffusion Models, inspired by various sources.☆15Nov 2, 2023Updated 2 years ago
- mask2former psg☆22Dec 12, 2022Updated 3 years ago
- ☆33Aug 9, 2024Updated last year
- ☆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☆426Dec 18, 2024Updated last year
- 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.
- The official implementation of Diffusion-KTO: Aligning Diffusion Models by Optimizing Human Utility☆69Aug 16, 2025Updated 11 months ago
- 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☆103Dec 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,899Dec 24, 2024Updated last year
- Cool extensions built for the nbdev framework☆14Jun 6, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Train high-quality text-to-image diffusion models in a data & compute efficient manner☆515Jun 25, 2026Updated 3 weeks ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,753Jan 8, 2024Updated 2 years ago
- ✨ Hotshot-XL: State-of-the-art AI text-to-GIF model trained to work alongside Stable Diffusion XL☆1,109Jan 23, 2024Updated 2 years ago
- Official implementation of Würstchen: Efficient Pretraining of Text-to-Image Models☆555Apr 6, 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…☆480Oct 21, 2024Updated last year
- 🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch - fork with video pseudo3d☆99Mar 5, 2023Updated 3 years ago
- End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).☆399Jan 8, 2026Updated 6 months ago