diff --git a/requirements-common.txt b/requirements-common.txt index 7051ca8cb50c..e5248572ce4d 100644 --- a/requirements-common.txt +++ b/requirements-common.txt @@ -5,7 +5,7 @@ requests >= 2.26.0 tqdm blake3 py-cpuinfo -transformers >= 4.45.2 # Required for Llama 3.2 and Qwen2-VL. +transformers >= 4.48.2 # Required for Bamba. tokenizers >= 0.19.1 # Required for Llama 3. protobuf # Required by LlamaTokenizer. fastapi >= 0.107.0, < 0.113.0; python_version < '3.9' diff --git a/requirements-test.in b/requirements-test.in index 13ad17b25673..229d743ec802 100644 --- a/requirements-test.in +++ b/requirements-test.in @@ -27,7 +27,7 @@ matplotlib # required for qwen-vl test mistral_common[opencv] >= 1.5.0 # required for pixtral test datamodel_code_generator # required for minicpm3 test lm-eval[api]==0.4.4 # required for model evaluation test - +transformers==4.48.2 # quantization bitsandbytes>=0.45.0 buildkite-test-collector==0.1.9 diff --git a/requirements-test.txt b/requirements-test.txt index df7e904bb0d3..e032aac710dd 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -2,7 +2,7 @@ # This file is autogenerated by pip-compile with Python 3.12 # by the following command: # -# python3.12 -m piptools compile requirements-test.in -o requirements-test.txt +# python3.12 -m piptools compile requirements-test.in -o requirements-test.txt # absl-py==2.1.0 # via rouge-score @@ -617,8 +617,9 @@ tqdm==4.66.6 # transformers tqdm-multiprocess==0.0.11 # via lm-eval -transformers==4.47.0 +transformers==4.48.2 # via + # -r requirements-test.in # genai-perf # lm-eval # peft