jiangyzy commited on
Commit
51de469
·
verified ·
1 Parent(s): 5b469a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ def prepare_data(device, input_image, x0, y0, x1, y1, polar, azimuth, text):
129
  return batch
130
 
131
 
132
- @spaces.GPU
133
  @torch.no_grad()
134
  def run_generation(sampler, model, device, input_image, x0, y0, x1, y1, polar, azimuth, text, seed):
135
  seed_everything(seed)
 
129
  return batch
130
 
131
 
132
+ @spaces.GPU(enable_queue=True)
133
  @torch.no_grad()
134
  def run_generation(sampler, model, device, input_image, x0, y0, x1, y1, polar, azimuth, text, seed):
135
  seed_everything(seed)