Update README.md

README.md

@@ -14,16 +14,10 @@ Base Model: https://huggingface.co/stabilityai/stable-diffusion-2-1
 
 
 ```python
-from diffusers import DiffusionPipeline
-from diffusers.utils import make_image_grid
-import torch
-
-pipe = DiffusionPipeline.from_pretrained(
-    "stabilityai/sd-turbo", 
-    # "stabilityai/stable-diffusion-2-1",
-    torch_dtype=torch.float16, variant="fp16"
-)
-pipe.load_lora_weights("stable-diffusion-2-1-DPO-LoRA")
+
+try:
+pipe.load_lora_weights("radames/stable-diffusion-2-1-DPO-LoRA", adapter_name="dpo-lora-sd21")
+pipe.set_adapters(["dpo-lora-sd21"], adapter_weights=[1.0]) # you can play with adapter_weights to increase the effect of the LoRA model
 seed = 123123
 pipe.to("cuda")
 prompt = "portrait headshot professional of elon musk"
@@ -42,10 +36,13 @@ images = pipe(
 make_image_grid(images, 1, 4)
 ```
 
-## Example
+## Examples
+
+Left Withoud DPO right with DPO LoRA
 
-Without DPO Lora  
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/3o1UBMgrBTT5SFfgYFmaq.png)
+<img src=https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/R8E0hRpWIE6OhhtvgJeEU.png style="max-width: 60rem;">
+<img src=https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/Eg4LbyxCfhmsk2INzqODw.png style="max-width: 60rem;">
+<img src=https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/GD7KumSCNweBWMJ1TArI-.png style="max-width: 60rem;">
+<img src=https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/SO7QoA9lZJY9hI0U4fBLy.png style="max-width: 60rem;">
+<img src=https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/ZWbQwIQ5OklEgF9RW581R.png style="max-width: 60rem;">
 
-With DPO Lora
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/6064e095abd8d3692e3e2ed6/chu29lik7TVGEyVddWyqb.png)