update bibtex
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ library_name: diffusers
|
|
| 9 |
---
|
| 10 |
|
| 11 |
<h1 align="center">
|
| 12 |
-
Ranking-based Preference Optimization </br> for Diffusion Models from Implicit User Feedback
|
| 13 |
</h1>
|
| 14 |
|
| 15 |
We present a learning framework that aligns text-to-image diffusion models with human preferences through inverse reinforcement learning and a balance of offline and online training.
|
|
@@ -18,9 +18,6 @@ We present a learning framework that aligns text-to-image diffusion models with
|
|
| 18 |
```python
|
| 19 |
import torch
|
| 20 |
from diffusers import StableDiffusionPipeline, UNet2DConditionModel
|
| 21 |
-
from dotenv import load_dotenv
|
| 22 |
-
|
| 23 |
-
load_dotenv()
|
| 24 |
|
| 25 |
unet = UNet2DConditionModel.from_pretrained(
|
| 26 |
"ylwu/diffusion-dro-sd1.5",
|
|
@@ -39,5 +36,18 @@ image = pipe(prompt).images[0]
|
|
| 39 |
image.save("example.png")
|
| 40 |
```
|
| 41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
## License
|
| 43 |
The model is licensed under the [CreativeML Open RAIL-M License](https://huggingface.co/spaces/CompVis/stable-diffusion-license).
|
|
|
|
| 9 |
---
|
| 10 |
|
| 11 |
<h1 align="center">
|
| 12 |
+
[NeurIPS 2025] Ranking-based Preference Optimization </br> for Diffusion Models from Implicit User Feedback
|
| 13 |
</h1>
|
| 14 |
|
| 15 |
We present a learning framework that aligns text-to-image diffusion models with human preferences through inverse reinforcement learning and a balance of offline and online training.
|
|
|
|
| 18 |
```python
|
| 19 |
import torch
|
| 20 |
from diffusers import StableDiffusionPipeline, UNet2DConditionModel
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
unet = UNet2DConditionModel.from_pretrained(
|
| 23 |
"ylwu/diffusion-dro-sd1.5",
|
|
|
|
| 36 |
image.save("example.png")
|
| 37 |
```
|
| 38 |
|
| 39 |
+
## Citation
|
| 40 |
+
```
|
| 41 |
+
@misc{wu2025rankingbasedpreferenceoptimizationdiffusion,
|
| 42 |
+
title={Ranking-based Preference Optimization for Diffusion Models from Implicit User Feedback},
|
| 43 |
+
author={Yi-Lun Wu and Bo-Kai Ruan and Chiang Tseng and Hong-Han Shuai},
|
| 44 |
+
year={2025},
|
| 45 |
+
eprint={2510.18353},
|
| 46 |
+
archivePrefix={arXiv},
|
| 47 |
+
primaryClass={cs.CV},
|
| 48 |
+
url={https://arxiv.org/abs/2510.18353},
|
| 49 |
+
}
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
## License
|
| 53 |
The model is licensed under the [CreativeML Open RAIL-M License](https://huggingface.co/spaces/CompVis/stable-diffusion-license).
|