Jeff850 commited on
Commit
a782412
1 Parent(s): fd67e37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ MAX_SEED = np.iinfo(np.int32).max
20
  MAX_IMAGE_SIZE = 2048
21
  DEFAULT_INFERENCE_STEPS = 4
22
 
23
- @spaces.GPU()
24
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_images=1, progress=gr.Progress(track_tqdm=True)):
25
  if randomize_seed:
26
  seed = random.randint(0, MAX_SEED)
 
20
  MAX_IMAGE_SIZE = 2048
21
  DEFAULT_INFERENCE_STEPS = 4
22
 
23
+ @spaces.GPU(duration=90)
24
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_images=1, progress=gr.Progress(track_tqdm=True)):
25
  if randomize_seed:
26
  seed = random.randint(0, MAX_SEED)