diff --git a/examples/llm_ptq/requirements-t5.txt b/examples/llm_ptq/requirements-t5.txt index 0347135464..d1d094caec 100644 --- a/examples/llm_ptq/requirements-t5.txt +++ b/examples/llm_ptq/requirements-t5.txt @@ -1 +1 @@ -transformers==4.48.0 +transformers==4.53.0 diff --git a/examples/vlm_ptq/requirements-vila.txt b/examples/vlm_ptq/requirements-vila.txt index 7391a5f268..9ee2fcd9c1 100644 --- a/examples/vlm_ptq/requirements-vila.txt +++ b/examples/vlm_ptq/requirements-vila.txt @@ -1,3 +1,3 @@ deepspeed>=0.16.0 git+https://github.com/bfshi/scaling_on_scales.git -transformers<=4.50.0 +transformers<=5.4.0 diff --git a/examples/windows/onnx_ptq/genai_llm/requirements.txt b/examples/windows/onnx_ptq/genai_llm/requirements.txt index b50e200a0b..b72fd8ebd3 100644 --- a/examples/windows/onnx_ptq/genai_llm/requirements.txt +++ b/examples/windows/onnx_ptq/genai_llm/requirements.txt @@ -1,4 +1,4 @@ datasets>=2.14.5 -onnx==1.19.0 -torch==2.9.0 -transformers==4.57.3 +onnx==1.21.0rc1 +torch==2.11.0 +transformers==5.4.0 diff --git a/examples/windows/onnx_ptq/sam2/requirements.txt b/examples/windows/onnx_ptq/sam2/requirements.txt index 1329ad6d85..34c59b336e 100644 --- a/examples/windows/onnx_ptq/sam2/requirements.txt +++ b/examples/windows/onnx_ptq/sam2/requirements.txt @@ -1,3 +1,3 @@ -onnx==1.17.0 +onnx==1.21.0rc1 onnxruntime-gpu==1.20.1 opencv-python diff --git a/examples/windows/onnx_ptq/whisper/requirements.txt b/examples/windows/onnx_ptq/whisper/requirements.txt index 4bf2fb98e5..4dd8968d63 100644 --- a/examples/windows/onnx_ptq/whisper/requirements.txt +++ b/examples/windows/onnx_ptq/whisper/requirements.txt @@ -4,10 +4,10 @@ datasets==2.19.0 evaluate jiwer librosa -onnx==1.19.0 +onnx==1.21.0rc1 onnxruntime-gpu==1.23.2 optimum==1.23.3 soundfile -torch==2.7.0+cu128 +torch==2.8.0 torchaudio==2.7.0+cu128 -transformers==4.57.3 +transformers==5.4.0 diff --git a/experimental/dms/pyproject.toml b/experimental/dms/pyproject.toml index 6c2f1f0d71..49ea079d82 100644 --- a/experimental/dms/pyproject.toml +++ b/experimental/dms/pyproject.toml @@ -9,7 +9,7 @@ requires-python = ">=3.10,<3.14" description = "Dynamic Memory Sparsification (DMS) for KV cache compression" # NOTE: Run `uv lock --upgrade` in this directory to update the dependencies for vulnerability scanning. dependencies = [ - "transformers==4.57.3", + "transformers==5.4.0", "datasets==4.4.2", "accelerate==1.4.0", "lm_eval[ruler]", diff --git a/pyproject.toml b/pyproject.toml index 96490dff0a..4f4e428643 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,7 +58,7 @@ onnx = [ "lief", "ml_dtypes", "onnx-graphsurgeon", - "onnx~=1.19.0", + "onnx>=1.19,<1.22", "onnxconverter-common~=1.16.0", # ORT for Windows "onnxruntime-gpu==1.22.0; platform_system == 'Windows'", @@ -81,7 +81,7 @@ hf = [ "nltk", "peft>=0.17.0", "sentencepiece>=0.2.1", # Also implicitly used in test_unified_export_megatron, test_vllm_fakequant_megatron_export - "transformers>=4.56,<5.0", # Should match modelopt/torch/__init__.py and tox.ini + "transformers>=4.56,<6.0", # Should match modelopt/torch/__init__.py and tox.ini "wonderwords", ] dev-lint = [