From 630e40dd134c2bfd1fb1344e63a5022d2a6685a6 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 9 Apr 2025 02:53:00 -0400
Subject: [PATCH] upgrade transformers to 4.51.1 (#2508)

* upgrade transformers to 4.51.1

* multigpu longer timeout
---
 cicd/multigpu.py | 2 +-
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/cicd/multigpu.py b/cicd/multigpu.py
index 453e8daee..66d4d9990 100644
--- 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,
diff --git a/requirements.txt b/requirements.txt
index 3a839d8a9..e8377b880 100644
--- 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