Unverified Commit 0035f1ce authored by Swipe4057's avatar Swipe4057 Committed by GitHub
Browse files

fix env flashinfer (#10910)

parent 5e21d6ae
...@@ -70,7 +70,7 @@ srt = [ ...@@ -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 = [ ...@@ -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",
] ]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment