amgad59 commited on
Commit
5e450cb
1 Parent(s): 3516167

Update app.py

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