From 0d99adb715a536558d16baca978fce81e27f61c1 Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Sat, 5 Apr 2025 14:20:23 -0700 Subject: [PATCH] upgrade transformers 4.51.0 (#5088) --- python/pyproject.toml | 2 +- scripts/ci_install_dependency.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/pyproject.toml b/python/pyproject.toml index c4f24303e..6504d4320 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -38,7 +38,7 @@ runtime_common = [ "pyzmq>=25.1.2", "soundfile==0.13.1", "torchao>=0.7.0", - "transformers==4.50.0", + "transformers==4.51.0", "uvicorn", "uvloop", "compressed-tensors", diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh index 8934c82fe..ec646b804 100755 --- a/scripts/ci_install_dependency.sh +++ b/scripts/ci_install_dependency.sh @@ -27,7 +27,7 @@ pip install -e "python[all]" --find-links ${FLASHINFER_REPO} # Install additional dependencies pip install torch_memory_saver -pip install transformers==4.50.0 sentence_transformers accelerate==1.4.0 peft pandas datasets timm torchaudio +pip install transformers==4.51.0 sentence_transformers accelerate==1.4.0 peft pandas datasets timm torchaudio # For compling xgrammar kernels pip install cuda-python nvidia-cuda-nvrtc-cu12