Kohaku-Blueleaf
commited on
Commit
·
4393fe6
1
Parent(s):
fdedf25
set duration
Browse files
app.py
CHANGED
|
@@ -193,7 +193,7 @@ kgen_models.load_model(tipo_model_name, device="cuda")
|
|
| 193 |
print("Models loaded successfully. UI is ready.")
|
| 194 |
|
| 195 |
|
| 196 |
-
@GPU
|
| 197 |
@torch.no_grad()
|
| 198 |
def generate(
|
| 199 |
nl_prompt: str,
|
|
|
|
| 193 |
print("Models loaded successfully. UI is ready.")
|
| 194 |
|
| 195 |
|
| 196 |
+
@GPU(duration=5)
|
| 197 |
@torch.no_grad()
|
| 198 |
def generate(
|
| 199 |
nl_prompt: str,
|