Update README.md

README.md

@@ -76,8 +76,10 @@ A **single step** is enough to generate high quality images.
 ```py
 from diffusers import AutoPipelineForText2Image
 import torch
+
 pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo", torch_dtype=torch.float16, variant="fp16")
 pipe.to("cuda")
+
 prompt = "A cinematic shot of a baby racoon wearing an intricate italian priest robe."
 image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
 ```
@@ -91,9 +93,14 @@ below.
 ```py
 from diffusers import AutoPipelineForImage2Image
 from diffusers.utils import load_image
+import torch
+
 pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sd-turbo", torch_dtype=torch.float16, variant="fp16")
+pipe.to("cuda")
+
 init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png").resize((512, 512))
 prompt = "cat wizard, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney, 8k"
+
 image = pipe(prompt, image=init_image, num_inference_steps=2, strength=0.5, guidance_scale=0.0).images[0]
 ```