diff --git a/docker/Dockerfile.nightly_torch b/docker/Dockerfile.nightly_torch index 53b8ccd804924..8d43de77aad59 100644 --- a/docker/Dockerfile.nightly_torch +++ b/docker/Dockerfile.nightly_torch @@ -312,4 +312,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \ # Logging to confirm the torch versions RUN pip freeze | grep -E 'torch|xformers|vllm|flashinfer' +# Logging to confirm all the packages are installed +RUN pip freeze + #################### UNITTEST IMAGE ############################# diff --git a/requirements/nightly_torch_test.txt b/requirements/nightly_torch_test.txt index e9b466d3a82d6..3475ada9f4c96 100644 --- a/requirements/nightly_torch_test.txt +++ b/requirements/nightly_torch_test.txt @@ -9,7 +9,9 @@ pytest-shard pytest-timeout librosa # required by audio tests in entrypoints/openai -sentence-transformers +sentence-transformers # required for embedding tests +transformers==4.51.3 +transformers_stream_generator # required for qwen-vl test numba == 0.61.2; python_version > '3.9' # testing utils boto3 @@ -38,4 +40,7 @@ matplotlib # required for qwen-vl test # required for Multi-Modal Models Test (Standard) num2words # required for smolvlm test pqdm -timm # required for internvl test \ No newline at end of file +timm # required for internvl test + +schemathesis>=3.39.15 # Required for openai schema test. +mteb>=1.38.11, <2 # required for mteb test