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
0bd409cf
Unverified
Commit
0bd409cf
authored
Jul 31, 2025
by
Michael Goin
Committed by
GitHub
Jul 31, 2025
Browse files
Move flashinfer-python to optional extra `vllm[flashinfer]` (#21959)
Signed-off-by:
mgoin
<
mgoin64@gmail.com
>
parent
e360316a
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
requirements/cuda.txt
requirements/cuda.txt
+1
-3
setup.py
setup.py
+3
-1
No files found.
requirements/cuda.txt
View file @
0bd409cf
...
@@ -12,5 +12,3 @@ torchaudio==2.7.1
...
@@ -12,5 +12,3 @@ torchaudio==2.7.1
torchvision==0.22.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
torchvision==0.22.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
# https://github.com/facebookresearch/xformers/releases/tag/v0.0.31
# https://github.com/facebookresearch/xformers/releases/tag/v0.0.31
xformers==0.0.31; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.7
xformers==0.0.31; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.7
# FlashInfer should be updated together with the Dockerfile
flashinfer_python==0.2.9rc2
\ No newline at end of file
setup.py
View file @
0bd409cf
...
@@ -671,7 +671,9 @@ setup(
...
@@ -671,7 +671,9 @@ setup(
[
"runai-model-streamer >= 0.13.3"
,
"runai-model-streamer-s3"
,
"boto3"
],
[
"runai-model-streamer >= 0.13.3"
,
"runai-model-streamer-s3"
,
"boto3"
],
"audio"
:
[
"librosa"
,
"soundfile"
,
"audio"
:
[
"librosa"
,
"soundfile"
,
"mistral_common[audio]"
],
# Required for audio processing
"mistral_common[audio]"
],
# Required for audio processing
"video"
:
[]
# Kept for backwards compatibility
"video"
:
[],
# Kept for backwards compatibility
# FlashInfer should be updated together with the Dockerfile
"flashinfer"
:
[
"flashinfer-python==0.2.9rc2"
],
},
},
cmdclass
=
cmdclass
,
cmdclass
=
cmdclass
,
package_data
=
package_data
,
package_data
=
package_data
,
...
...
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