Spaces:
Running
on
Zero
Running
on
Zero
Add Zero60 to 120s
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ if torch.cuda.is_available():
|
|
| 34 |
tokenizer.use_default_system_prompt = False
|
| 35 |
|
| 36 |
|
| 37 |
-
@spaces.GPU
|
| 38 |
def generate(
|
| 39 |
message: str,
|
| 40 |
chat_history: list[tuple[str, str]],
|
|
|
|
| 34 |
tokenizer.use_default_system_prompt = False
|
| 35 |
|
| 36 |
|
| 37 |
+
@spaces.GPU(duration=120)
|
| 38 |
def generate(
|
| 39 |
message: str,
|
| 40 |
chat_history: list[tuple[str, str]],
|