Added sample code

README.md

@@ -19,6 +19,31 @@ If you want more details on how to generate your own blip cpationed dataset see
 
 Training was done using a slightly modified version of Hugging-Face's text to image training [example script](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image.py)
 
+```py
+from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
+import torch
+
+# this will substitute the default PNDM scheduler for K-LMS  
+lms = LMSDiscreteScheduler(
+    beta_start=0.00085, 
+    beta_end=0.012, 
+    beta_schedule="scaled_linear"
+)
+
+guidance_scale=9
+seed=7777
+steps=100
+
+model_id = "Norod78/sd-simpsons-model"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=lms, torch_dtype=torch.float16)
+pipe.to("cuda")
+
+prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]  
+    
+image.save("astronaut_rides_horse.png")
+```
+
 ## About
 
 Put in a text prompt and generate cartoony/simpsony images