xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-31 17:17:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add logging for torch nightly version (#17669)

Browse Source 
Signed-off-by: Yang Wang <elainewy@meta.com>
This commit is contained in:
Yang Wang 2025-05-06 17:45:51 -07:00 committed by GitHub
parent ed3a1d2106
commit 6de3e13413
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.nightly_torch
View File

@ -309,5 +309,7 @@ ENV HF_HUB_ENABLE_HF_TRANSFER 1
RUN --mount=type=cache,target=/root/.cache/uv \ RUN --mount=type=cache,target=/root/.cache/uv \
uv pip install --system -r requirements/nightly_torch_test.txt uv pip install --system -r requirements/nightly_torch_test.txt
#################### UNITTEST IMAGE ############################# # Logging to confirm the torch versions
RUN pip freeze | grep -E 'torch|xformers|vllm|flashinfer'
#################### UNITTEST IMAGE #############################

10
requirements/nightly_torch_test.txt
View File

@ -8,7 +8,6 @@ pytest-rerunfailures
pytest-shard 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
numba == 0.61.2; python_version > '3.9' numba == 0.61.2; python_version > '3.9'
@ -31,3 +30,12 @@ bitsandbytes>=0.45.3
# required for minicpmo_26 test # required for minicpmo_26 test
vector_quantize_pytorch vector_quantize_pytorch
vocos vocos
# required for Basic Models Test
blobfile # required for kimi-vl test
matplotlib # required for qwen-vl test
# required for Multi-Modal Models Test (Standard)
num2words # required for smolvlm test
pqdm
timm # required for internvl test