Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
| 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
|