Update README.md

README.md

@@ -15,11 +15,19 @@ Base Model: https://huggingface.co/stabilityai/stable-diffusion-2-1
 
 ```python
 
-try:
+from diffusers import DiffusionPipeline
+from diffusers.utils import make_image_grid
+import torch
+
+pipe = DiffusionPipeline.from_pretrained(
+    "stabilityai/sd-turbo",  # SD Turbo is a destilled version of Stable Diffusion 2.1
+    # "stabilityai/stable-diffusion-2-1", # for the original stable diffusion 2.1 model
+    torch_dtype=torch.float16, variant="fp16"
+)
+pipe.to("cuda")
 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"
 negative_prompt = "3d render, cartoon, drawing, art, low light"
 generator = torch.Generator().manual_seed(seed)