Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
35f1cc3
1
Parent(s):
5f08391
Update app.py
Browse files
app.py
CHANGED
|
@@ -83,7 +83,7 @@ def encode_prompt(
|
|
| 83 |
return prompt_embeds, prompt_masks
|
| 84 |
|
| 85 |
|
| 86 |
-
@spaces.GPU
|
| 87 |
def load_model(args, master_port, rank):
|
| 88 |
# import here to avoid huggingface Tokenizer parallelism warnings
|
| 89 |
from diffusers.models import AutoencoderKL
|
|
|
|
| 83 |
return prompt_embeds, prompt_masks
|
| 84 |
|
| 85 |
|
| 86 |
+
@spaces.GPU(duration=200)
|
| 87 |
def load_model(args, master_port, rank):
|
| 88 |
# import here to avoid huggingface Tokenizer parallelism warnings
|
| 89 |
from diffusers.models import AutoencoderKL
|