Update README.md

README.md

@@ -20,18 +20,23 @@ This repository is based on [Diffusers](https://huggingface.co/docs/diffusers/in
 Load StableDiffusionPAGPipeline as below:
 
 ```
+from diffusers import StableDiffusionPipeline
+
 pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     custom_pipeline="hyoungwoncho/sd_perturbed_attention_guidance",
     torch_dtype=torch.float16,
     safety_checker=None
 )
+
+device="cuda"
+pipe = pipe.to(device)
 ```
 
 Sampling:
 
 ```
-output_baseline = pipe(
+output = pipe(
         prompts,
         width=512,
         height=512,