diff --git a/docker/Dockerfile.nightly_torch b/docker/Dockerfile.nightly_torch index 6989106c42965..53b8ccd804924 100644 --- a/docker/Dockerfile.nightly_torch +++ b/docker/Dockerfile.nightly_torch @@ -309,5 +309,7 @@ ENV HF_HUB_ENABLE_HF_TRANSFER 1 RUN --mount=type=cache,target=/root/.cache/uv \ uv pip install --system -r requirements/nightly_torch_test.txt -#################### UNITTEST IMAGE ############################# +# Logging to confirm the torch versions +RUN pip freeze | grep -E 'torch|xformers|vllm|flashinfer' +#################### UNITTEST IMAGE ############################# diff --git a/requirements/nightly_torch_test.txt b/requirements/nightly_torch_test.txt index e2711354ac10e..3aebcaa623c03 100644 --- a/requirements/nightly_torch_test.txt +++ b/requirements/nightly_torch_test.txt @@ -8,7 +8,6 @@ pytest-rerunfailures pytest-shard pytest-timeout - librosa # required by audio tests in entrypoints/openai sentence-transformers numba == 0.61.2; python_version > '3.9' @@ -31,3 +30,12 @@ bitsandbytes>=0.45.3 # required for minicpmo_26 test vector_quantize_pytorch vocos + +# required for Basic Models Test +blobfile # required for kimi-vl test +matplotlib # required for qwen-vl test + +# required for Multi-Modal Models Test (Standard) +num2words # required for smolvlm test +pqdm +timm # required for internvl test