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 \ ...@@ -312,4 +312,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
# Logging to confirm the torch versions # Logging to confirm the torch versions
RUN pip freeze | grep -E 'torch|xformers|vllm|flashinfer' RUN pip freeze | grep -E 'torch|xformers|vllm|flashinfer'
# Logging to confirm all the packages are installed
RUN pip freeze
#################### UNITTEST IMAGE ############################# #################### UNITTEST IMAGE #############################
...@@ -9,7 +9,9 @@ pytest-shard ...@@ -9,7 +9,9 @@ pytest-shard
pytest-timeout pytest-timeout
librosa # required by audio tests in entrypoints/openai 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' numba == 0.61.2; python_version > '3.9'
# testing utils # testing utils
boto3 boto3
...@@ -39,3 +41,6 @@ matplotlib # required for qwen-vl test ...@@ -39,3 +41,6 @@ matplotlib # required for qwen-vl test
num2words # required for smolvlm test num2words # required for smolvlm test
pqdm 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
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