Spaces:
Sleeping
Sleeping
hoduyquocbao
commited on
Commit
•
502b366
1
Parent(s):
999b0b0
edit time
Browse files
app.py
CHANGED
@@ -275,7 +275,7 @@ def handle_functions(function_call: Dict[str, Any], prompt: str, chat_history: L
|
|
275 |
|
276 |
# ---------------------------- Giao Diện Gradio ---------------------------- #
|
277 |
|
278 |
-
@spaces.GPU(duration=
|
279 |
def generate(
|
280 |
message: str,
|
281 |
chat_history: List[Tuple[str, str]],
|
@@ -458,7 +458,7 @@ class SaveCheckpointCallback(TrainerCallback):
|
|
458 |
return control # Trả về đối tượng control hiện tại
|
459 |
|
460 |
# Định Nghĩa Hàm Huấn Luyện với Decorator @spaces.GPU
|
461 |
-
@spaces.GPU(duration=
|
462 |
def run_training():
|
463 |
"""
|
464 |
Hàm huấn luyện mô hình sử dụng GPU với thời gian hạn chế.
|
|
|
275 |
|
276 |
# ---------------------------- Giao Diện Gradio ---------------------------- #
|
277 |
|
278 |
+
@spaces.GPU(duration=60, queue=False)
|
279 |
def generate(
|
280 |
message: str,
|
281 |
chat_history: List[Tuple[str, str]],
|
|
|
458 |
return control # Trả về đối tượng control hiện tại
|
459 |
|
460 |
# Định Nghĩa Hàm Huấn Luyện với Decorator @spaces.GPU
|
461 |
+
@spaces.GPU(duration=60, queue=False)
|
462 |
def run_training():
|
463 |
"""
|
464 |
Hàm huấn luyện mô hình sử dụng GPU với thời gian hạn chế.
|