modified code example

README.md

@@ -30,16 +30,18 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 pipe = pipe.to("cuda")
 
 prompt = "a classroom"
-image = pipe(prompt).images[0]
+image = pipe(prompt, height=512, width=512).images[0]
+# ^ alternatively you can use for height=1920, width=1080 if you have enough CUDA memory
+# make sure both height and width are divisible by 8
 image.save("./classroom.png")
 
 prompt = "a hospital building, two trees"
-image = pipe(prompt).images[0]
+image = pipe(prompt, height=512, width=512).images[0]
 
 image.save("./hospital.png")
 
 prompt = "a street at night with nobody around"
-image = pipe(prompt).images[0]
+image = pipe(prompt, height=512, width=512).images[0]
 
 image.save("./nightstreet.png")
 ```