[Torch Nightly]add missing dependency (#18770)

Signed-off-by: Yang Wang <elainewy@meta.com>
This commit is contained in:
Yang Wang
2025-06-04 21:56:12 -07:00
committed by GitHub
parent a408820f2f
commit 25b918eee6
2 changed files with 10 additions and 2 deletions

View File

@ -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 #############################

View File

@ -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
timm # required for internvl test
schemathesis>=3.39.15 # Required for openai schema test.
mteb>=1.38.11, <2 # required for mteb test