Update README.md

README.md

@@ -42,30 +42,28 @@ pip install invisible_watermark transformers accelerate safetensors
 
 Running the pipeline (if you don't swap the scheduler it will run with the default **EulerDiscreteScheduler** in this example we are swapping it to **EulerAncestralDiscreteScheduler**:
 ```py
-import torch
-from torch import autocast
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
-model = "hotshotco/SDXL-512"
+
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    model, 
-    torch_dtype=torch.float16, 
-    use_safetensors=True, 
-    variant="fp16"
-    )
+    "hotshotco/SDXL-512",
+    use_safetensors=True,
+).to('cuda')
+
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-pipe.to('cuda')
+
 prompt = "a woman laughing"
 negative_prompt = ""
+
 image = pipe(
-    prompt, 
-    negative_prompt=negative_prompt, 
+    prompt,
+    negative_prompt=negative_prompt,
     width=512,
     height=512,
-    guidance_scale=12,
-    target_size=(1024,1024),
-    original_size=(4096,4096),
+    target_size=(1024, 1024),
+    original_size=(4096, 4096),
     num_inference_steps=50
-    ).images[0]
+).images[0]
+
 image.save("woman_laughing.png")
 ```