Update README.md

README.md

@@ -74,19 +74,20 @@ please have a look at the [Stable Diffusion](https://huggingface.co/docs/diffuse
 You can also export the model to [ONNX](https://huggingface.co/docs/diffusers/optimization/onnx), [MPS](https://huggingface.co/docs/diffusers/optimization/mps) and/or [FLAX/JAX]().
 
 ```python
-from diffusers import StableDiffusionPipeline
 import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
 
-model_id = "TheRafal/everything-v1"
-branch_name = "main"
-
-pipe = StableDiffusionPipeline.from_pretrained(model_id, revision=branch_name, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained(
+    'TheRafal/everything-v1',
+    torch_dtype=torch.float32
+).to('cuda')
 
 prompt = "1girl, monika \(doki doki literature club\)"
-image = pipe(prompt).images[0]
-
-image.save("./monika.png")
+with autocast("cuda"):
+    image = pipe(prompt, guidance_scale=6)["sample"][0]  
+    
+image.save("test.png")
 ```
 
 ----