[Torch Nightly]add missing dependency (#18770)
Signed-off-by: Yang Wang <elainewy@meta.com>
This commit is contained in:
@ -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
|
||||
@ -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
|
||||
|
Reference in New Issue
Block a user