zacharyhorvitz / Fk-Diffusion-Steering
A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
☆112Updated last month
Alternatives and similar repositories for Fk-Diffusion-Steering:
Users that are interested in Fk-Diffusion-Steering are comparing it to the libraries listed below
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆101Updated 5 months ago
- ☆51Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆67Updated 3 weeks ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 8 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Framework for Cross-Modality Evol…☆142Updated last week
- Code for the paper DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latents, ICML 2024☆83Updated 9 months ago
- [ICLR2025] DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆117Updated this week
- ☆84Updated 5 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆223Updated 3 months ago
- [ICML 2024] Compositional Image Decomposition with Diffusion Models☆49Updated 8 months ago
- ☆28Updated 5 months ago
- ☆27Updated 10 months ago
- Focused on fast experimentation and simplicity☆69Updated 2 months ago
- [NeurIPS 2024] ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimization☆132Updated last month
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data"☆33Updated last month
- ☆149Updated 3 months ago
- 🦾 EvalGIM (pronounced as "EvalGym") is an evaluation library for generative image models. It enables easy-to-use, reproducible automatic…☆68Updated 3 months ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆50Updated last month
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆138Updated 5 months ago
- Official Implementation of understanding the latent space of diffusion models through the lens of riemannian geometry (NeurIPS 2023)☆84Updated last year
- NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…☆41Updated 4 months ago
- ☆68Updated 4 months ago
- My take on Flow Matching☆41Updated 2 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆201Updated last week
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆93Updated last year
- ☆30Updated 2 months ago