Add model card information
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
@@ -1 +1,25 @@
|
|
1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
pipeline_tag: image-to-image
|
3 |
+
library_name: diffusers
|
4 |
+
license: mit # Assuming MIT license. Please verify and update if incorrect.
|
5 |
+
---
|
6 |
+
|
7 |
+
# FashionDPO: Fashion Image Generation with Direct Preference Optimization
|
8 |
+
|
9 |
+
This repository provides FashionDPO models, including checkpoints fine-tuned on the iFashion and Polyvore datasets, and the trained VBPR model. FashionDPO utilizes a novel approach to fashion image generation by incorporating direct preference optimization. This allows for the generation of high-quality fashion images that align with user preferences. The model is based on [this paper](https://huggingface.co/papers/2504.12900).
|
10 |
+
|
11 |
+
## Usage:
|
12 |
+
|
13 |
+
The FashionDPO pipeline can be used to generate fashion images. Refer to the [Github repository](https://github.com/Yzcreator/FashionDPO) for detailed usage instructions, including how to generate initial recommendations and incorporate feedback from multiple experts. This includes instructions for running `sample.py` for image generation and `multiple_evaluate.py` for feedback generation.
|
14 |
+
|
15 |
+
## Model Checkpoints:
|
16 |
+
|
17 |
+
- `checkpoint_ifashion`: Fine-tuned on the iFashion dataset.
|
18 |
+
- `checkpoint_polyvore`: Fine-tuned on the Polyvore dataset.
|
19 |
+
|
20 |
+
These checkpoints are available in the repository.
|
21 |
+
|
22 |
+
|
23 |
+
## Citation:
|
24 |
+
|
25 |
+
(Add citation here once the paper is published and available)
|