hoduyquocbao commited on
Commit
502b366
1 Parent(s): 999b0b0
Files changed (1) hide show
  1. app.py +2 -2
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=20, queue=False)
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=15, 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ế.
 
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ế.