Unverified Commit 25b918ee authored by Yang Wang's avatar Yang Wang Committed by GitHub
Browse files

[Torch Nightly]add missing dependency (#18770)


Signed-off-by: default avatarYang Wang <elainewy@meta.com>
parent a408820f
......@@ -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 #############################
......@@ -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
......@@ -39,3 +41,6 @@ matplotlib # required for qwen-vl test
num2words # required for smolvlm test
pqdm
timm # required for internvl test
schemathesis>=3.39.15 # Required for openai schema test.
mteb>=1.38.11, <2 # required for mteb test
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment