Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
25b918ee
Unverified
Commit
25b918ee
authored
Jun 04, 2025
by
Yang Wang
Committed by
GitHub
Jun 04, 2025
Browse files
[Torch Nightly]add missing dependency (#18770)
Signed-off-by:
Yang Wang
<
elainewy@meta.com
>
parent
a408820f
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
2 deletions
+10
-2
docker/Dockerfile.nightly_torch
docker/Dockerfile.nightly_torch
+3
-0
requirements/nightly_torch_test.txt
requirements/nightly_torch_test.txt
+7
-2
No files found.
docker/Dockerfile.nightly_torch
View file @
25b918ee
...
@@ -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 #############################
requirements/nightly_torch_test.txt
View file @
25b918ee
...
@@ -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
...
@@ -38,4 +40,7 @@ matplotlib # required for qwen-vl test
...
@@ -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
\ No newline at end of file
schemathesis>=3.39.15 # Required for openai schema test.
mteb>=1.38.11, <2 # required for mteb test
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment