Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ def chunk_generations(num_images):
|
|
21 |
"""Split number of images into batches of BATCH_SIZE"""
|
22 |
return [min(BATCH_SIZE, num_images - i) for i in range(0, num_images, BATCH_SIZE)]
|
23 |
|
24 |
-
@spaces.GPU(duration=
|
25 |
def generate_image(
|
26 |
prompt,
|
27 |
num_inference_steps=1,
|
|
|
21 |
"""Split number of images into batches of BATCH_SIZE"""
|
22 |
return [min(BATCH_SIZE, num_images - i) for i in range(0, num_images, BATCH_SIZE)]
|
23 |
|
24 |
+
@spaces.GPU(duration=25)
|
25 |
def generate_image(
|
26 |
prompt,
|
27 |
num_inference_steps=1,
|