CodeGoat24 commited on
Commit
fb52007
·
verified ·
1 Parent(s): c68a470

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -3
README.md CHANGED
@@ -1,3 +1,43 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ base_model:
4
+ - stabilityai/sdxl-turbo
5
+ ---
6
+
7
+ # sdxl-turbo-unified-reward-dpo
8
+
9
+ ## Model Summary
10
+
11
+ This model is trained on sdxl-turbo based on DPO preference data constructed by our [UnifiedReward-7B](https://huggingface.co/CodeGoat24/UnifiedReward-7b) for enhanced image generation quality.
12
+
13
+ For further details, please refer to the following resources:
14
+ - 📰 Paper:
15
+ - 🪐 Project Page: https://codegoat24.github.io/UnifiedReward/
16
+ - 🤗 Model Collections: https://huggingface.co/collections/CodeGoat24/unifiedreward-models-67c3008148c3a380d15ac63a
17
+ - 🤗 Dataset Collections: https://huggingface.co/collections/CodeGoat24/unifiedreward-training-data-67c300d4fd5eff00fa7f1ede
18
+ - 👋 Point of Contact: [Yibin Wang](https://codegoat24.github.io)
19
+
20
+
21
+ ### Quick Start
22
+
23
+ SDXL-Turbo does not make use of `guidance_scale` or `negative_prompt`, we disable it with `guidance_scale=0.0`.
24
+ Preferably, the model generates images of size 512x512 but higher image sizes work as well.
25
+ A **single step** is enough to generate high quality images.
26
+
27
+ ```py
28
+ from diffusers import AutoPipelineForText2Image
29
+ import torch
30
+
31
+ pipe = AutoPipelineForText2Image.from_pretrained("CodeGoat24/sdxl-turbo-unified-reward-dpo", torch_dtype=torch.float16, variant="fp16")
32
+ pipe.to("cuda")
33
+
34
+ prompt = "A cinematic shot of a baby racoon wearing an intricate italian priest robe."
35
+
36
+ image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
37
+ ```
38
+
39
+ ## Citation
40
+
41
+ ```
42
+
43
+ ```