biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Torch Nightly]add missing dependency (#18770)

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

3
docker/Dockerfile.nightly_torch
View File

@@ -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
requirements/nightly_torch_test.txt
View File

@@ -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
@@ -38,4 +40,7 @@ matplotlib # required for qwen-vl test
# required for Multi-Modal Models Test (Standard) # required for Multi-Modal Models Test (Standard)
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