Gijs Wijngaard
commited on
Commit
·
330538d
1
Parent(s):
6ab8a7f
feat: Specify GPU duration for chat_predict
Browse files
app.py
CHANGED
|
@@ -105,7 +105,7 @@ def _prepare_conversation(
|
|
| 105 |
conversation.append(_convert_new_message(message))
|
| 106 |
return conversation
|
| 107 |
|
| 108 |
-
@spaces.GPU
|
| 109 |
def chat_predict(
|
| 110 |
message: Dict[str, Any],
|
| 111 |
history: List[Dict[str, Any]],
|
|
|
|
| 105 |
conversation.append(_convert_new_message(message))
|
| 106 |
return conversation
|
| 107 |
|
| 108 |
+
@spaces.GPU(duration=120)
|
| 109 |
def chat_predict(
|
| 110 |
message: Dict[str, Any],
|
| 111 |
history: List[Dict[str, Any]],
|