Update README.md

README.md

@@ -41,7 +41,35 @@ Special VAE used for training: [madebyollin/sdxl-vae-fp16-fix](https://huggingfa
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
+from diffusers import DiffusionPipeline
+import torch
+
+# Load Stable Diffusion XL Base1.0
+pipe = DiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    torch_dtype=torch.float16,
+    variant="fp16",
+    use_safetensors=True
+).to("cuda")
+
+# Optional CPU offloading to save some GPU Memory
+pipe.enable_model_cpu_offload()
+
+# Loading Trained Old Book Illustrations LoRA Weights
+pipe.load_lora_weights("AdamLucek/sdxl-base-1.0-oldbookillustrations-lora")
+
+# Generate an Image
+prompt = "An Apple"
+
+image = pipe(
+    prompt = prompt,
+    num_inference_steps=50,
+    height=1024,
+    width=1024,
+).images[0]
+
+# Save the image
+image.save("SDXL_OldBookIllustrations.png")
 ```
 
 #### Limitations and bias