VamooseBambel commited on
Commit
1e26c4c
·
verified ·
1 Parent(s): 5e2c0e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=20)
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,