Update README.md

README.md

@@ -54,12 +54,25 @@ pipeline.scheduler = DDIMScheduler.from_config(pipeline.scheduler.config)
 
 prompt = ["a spiderman bayc nft"]
 neg_prompt = ["realistic,disfigured face,eye patch,disfigured eyes, disfigured, deformed,bad anatomy"] * len(prompt)
+num_samples = 3
+guidance_scale = 9
+num_inference_steps = 50
+height = 512
+width = 512
 
-# Other parameters like num_samples, guidance_scale, etc., are defined here.
+seed = np.random.randint(0, 2**20 - 1)
+print("Seed: {}".format(str(seed)))
+generator = torch.Generator(device='cuda').manual_seed(seed)
 
 with autocast("cuda"), torch.inference_mode():
     imgs = pipeline(
-        # Parameters like prompt, negative_prompt, etc., are passed here.
+        prompt,
+        negative_prompt=neg_prompt,
+        height=height, width=width,
+        num_images_per_prompt=num_samples,
+        num_inference_steps=num_inference_steps,
+        guidance_scale=guidance_scale,
+        generator=generator
     ).images
 
 for img in imgs: