Merge pull request #232 from hihaluemen/fix/gpu-ci-transformers-compat

[WS1][CI] Pin Transformers 5.13.1 for Torch 2.4 GPU CI

Author: Flink-ddd

ci/run_gpu_ci.sh

@@ -186,7 +186,7 @@ TORCH_INDEX_URL="${TORCH_INDEX_URL:-https://download.pytorch.org/whl/cu124}"
 # --no-build-isolation: torch must be visible to setup.py, else the extension is silently skipped.
 # --no-deps: keep the pinned torch; do not let the editable install re-resolve it.
 "$PY" -m pip install --no-build-isolation --no-deps -e .
-"$PY" -m pip install --no-cache-dir numpy tabulate accelerate transformers pytest
+"$PY" -m pip install --no-cache-dir numpy tabulate accelerate "transformers==5.13.1" pytest
 nvidia-smi
 # Fail fast if _C did not build or cannot launch, instead of silently using native fallbacks.
 "$PY" scripts/ci_smoke.py

pyproject.toml

@@ -17,7 +17,7 @@ dependencies = [
     "tabulate",
     "numpy",
     "accelerate",
-    "transformers",
+    "transformers==5.13.1",
 ]
 
 [project.optional-dependencies]

setup.py

@@ -179,7 +179,7 @@ def get_cmdclass():
         "tabulate",
         "numpy",
         "accelerate",
-        "transformers",
+        "transformers==5.13.1",
     ],
     ext_modules=get_extensions(),
     cmdclass=get_cmdclass(),