amgad59 commited on
Commit
62c6d26
1 Parent(s): b65eb59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -41,6 +41,7 @@ def generate_image_fn(prompt: str, unconditional_guidance_scale: int) -> list:
41
  prompt,
42
  batch_size=num_images_to_gen,
43
  unconditional_guidance_scale=unconditional_guidance_scale,
 
44
  )
45
  end_time = time.time()
46
  print(f"Time taken: {end_time - start_time} seconds.")
 
41
  prompt,
42
  batch_size=num_images_to_gen,
43
  unconditional_guidance_scale=unconditional_guidance_scale,
44
+ num_steps = 100,
45
  )
46
  end_time = time.time()
47
  print(f"Time taken: {end_time - start_time} seconds.")