Skip to content

Commit 630e40d

Browse files
authored
upgrade transformers to 4.51.1 (axolotl-ai-cloud#2508)
* upgrade transformers to 4.51.1 * multigpu longer timeout
1 parent bf9efe2 commit 630e40d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

cicd/multigpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def run_cmd(cmd: str, run_folder: str):
6868
@app.function(
6969
image=cicd_image,
7070
gpu=GPU_CONFIG,
71-
timeout=60 * 60,
71+
timeout=90 * 60,
7272
cpu=8.0,
7373
memory=131072 * N_GPUS,
7474
volumes=VOLUME_CONFIG,

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ liger-kernel==0.5.6
1212
packaging==23.2
1313

1414
peft==0.15.1
15-
transformers==4.51.0
15+
transformers==4.51.1
1616
tokenizers>=0.21.1
1717
accelerate==1.6.0
1818
datasets==3.5.0

0 commit comments

Comments
 (0)