Update README.md

README.md

@@ -93,22 +93,22 @@ pipe = pipe.to("cuda")
 def random_seed():
   return random.randint(0, 2**32 - 1)
 
-# what you want to see
-prompt = "((oil painting)), gentle waves, bright blue sky, white sails billowing, sun glistening on the surface, salty sea air, distant horizon, calm breeze, birds soaring overhead, vibrant colors, artstation digital painting, high resolution, uhd, 4 k, 8k wallpaper"
 
-# what you don't want to see
-negative_prompt = ["low-res", "blurry", "haze", "dark clouds looming", "choppy waves", "engine failing", "sails tattered", "stormy winds", "fear of the unknown"]
+prompt = "((oil painting)), gentle waves, bright blue sky, white sails billowing, sun glistening on the surface, salty sea air, distant horizon, calm breeze, birds soaring overhead, vibrant colors, artstation digital painting, high resolution, uhd, 4 k, 8k wallpaper"   # what you want to see
+negative_prompt = ["low-res", "blurry", "haze", "dark clouds looming", "choppy waves", "engine failing", "sails tattered", "stormy winds", "fear of the unknown"]   # what you don't want to see
+seed = random_seed()               # replace with the desired seed if needed
+width, height = 1280, 768          # width and height of the generated image
+cfg_scale = 7.5                    # classifer free guidance scale, 7 to 11 is usually a good range
+num_inference_steps = 40           # sampling steps
 
-# replace with the desired seed if needed
-seed = random_seed()
 
 generator = torch.Generator("cuda").manual_seed(seed)
 with torch.autocast("cuda"):
     image = pipe(prompt=prompt,
-                  num_inference_steps=40,
-                  height=768, width=1280,
+                  num_inference_steps=num_inference_steps,
+                  width=width, height=height,
                   generator=generator,
-                  guidance_scale=7.5).images[0]
+                  guidance_scale=cfg_scale).images[0]
 
 def generate_filename(string, seed):
     invalid_chars = ["<", ">", ":", '"', "/", "\\", "|", "?", "*"]