fix env flashinfer (#10910)

This commit is contained in:
Swipe4057
2025-09-26 01:44:48 +03:00
committed by GitHub
parent 5e21d6aec0
commit 0035f1cefa

View File

@@ -70,7 +70,7 @@ srt = [
"torchaudio==2.8.0", "torchaudio==2.8.0",
"torchvision", "torchvision",
"cuda-python", "cuda-python",
"flashinfer_python==0.3.1", "flashinfer_python==0.4.0rc1",
] ]
blackwell = [ blackwell = [
@@ -80,7 +80,7 @@ blackwell = [
"torchaudio==2.8.0", "torchaudio==2.8.0",
"torchvision", "torchvision",
"cuda-python", "cuda-python",
"flashinfer_python==0.3.1", "flashinfer_python==0.4.0rc1",
"nvidia-cutlass-dsl==4.2.0", "nvidia-cutlass-dsl==4.2.0",
] ]