fix env flashinfer (#10910)
This commit is contained in:
@@ -70,7 +70,7 @@ srt = [
|
||||
"torchaudio==2.8.0",
|
||||
"torchvision",
|
||||
"cuda-python",
|
||||
"flashinfer_python==0.3.1",
|
||||
"flashinfer_python==0.4.0rc1",
|
||||
]
|
||||
|
||||
blackwell = [
|
||||
@@ -80,7 +80,7 @@ blackwell = [
|
||||
"torchaudio==2.8.0",
|
||||
"torchvision",
|
||||
"cuda-python",
|
||||
"flashinfer_python==0.3.1",
|
||||
"flashinfer_python==0.4.0rc1",
|
||||
"nvidia-cutlass-dsl==4.2.0",
|
||||
]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user