Spaces:
Runtime error
Runtime error
Walmart-the-bag
commited on
Commit
•
87a7979
1
Parent(s):
fdcc962
more seconds
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ class StopOnTokens(StoppingCriteria):
|
|
19 |
return True
|
20 |
return False
|
21 |
|
22 |
-
@spaces.GPU(120)
|
23 |
def predict(message, history, temperature, max_tokens, top_p, top_k):
|
24 |
history_transformer_format = history + [[message, ""]]
|
25 |
stop = StopOnTokens()
|
|
|
19 |
return True
|
20 |
return False
|
21 |
|
22 |
+
@spaces.GPU(duration=120)
|
23 |
def predict(message, history, temperature, max_tokens, top_p, top_k):
|
24 |
history_transformer_format = history + [[message, ""]]
|
25 |
stop = StopOnTokens()
|