multigpu longer timeout

upgrade transformers to 4.51.1
2025-04-09 01:54:35 -04:00 · 2025-04-09 00:20:50 -04:00
2 changed files with 2 additions and 2 deletions
--- a/cicd/multigpu.py
+++ b/cicd/multigpu.py
@@ -68,7 +68,7 @@ def run_cmd(cmd: str, run_folder: str):
@app.function(
    image=cicd_image,
    gpu=GPU_CONFIG,
-    timeout=60 * 60,
+    timeout=90 * 60,
    cpu=8.0,
    memory=131072 * N_GPUS,
    volumes=VOLUME_CONFIG,
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,7 +12,7 @@ liger-kernel==0.5.6
 packaging==23.2

 peft==0.15.1
-transformers==4.51.0
+transformers==4.51.1
 tokenizers>=0.21.1
 accelerate==1.6.0
 datasets==3.5.0
Author	SHA1	Message	Date
Wing Lian	37a66e6866	multigpu longer timeout	2025-04-09 01:54:35 -04:00
Wing Lian	9f69597a5f	upgrade transformers to 4.51.1	2025-04-09 00:20:50 -04:00