pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("PLAN-Lab/RewardFlow", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]RewardFlow: Generate Images by Optimizing What You Reward
Accepted at CVPR 2026.
Citation
@inproceedings{rewardflow2026,
title = {RewardFlow: Generate Images by Optimizing What You Reward},
author = {Susladkar, Onkar Kishor and Jang, Dong-Hwan and Prakash, Tushar and Juvekar, Adheesh Sunil and Shah, Vedant and Barik, Ayush and Bashir, Nabeel and Wahed, Muntasir and Shrirao, Ritish and Lourentzou, Ismini},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2026}
}
- Downloads last month
- -
Model tree for PLAN-Lab/RewardFlow
Base model
black-forest-labs/FLUX.1-dev
# Gated model: Login with a HF token with gated access permission hf auth login