Dakerqi commited on
Commit
fc15728
·
verified ·
1 Parent(s): bc4c10a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def encode_prompt(prompt_batch, text_encoder, tokenizer, proportion_empty_prompt
88
 
89
  return prompt_embeds, prompt_masks
90
 
91
- @spaces.GPU(duration=200)
92
  @torch.no_grad()
93
  def model_main(args, master_port, rank, request_queue, response_queue, mp_barrier):
94
  # import here to avoid huggingface Tokenizer parallelism warnings
 
88
 
89
  return prompt_embeds, prompt_masks
90
 
91
+ @spaces.GPU(duration=20)
92
  @torch.no_grad()
93
  def model_main(args, master_port, rank, request_queue, response_queue, mp_barrier):
94
  # import here to avoid huggingface Tokenizer parallelism warnings