Update app.py
Browse files
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.")
|