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