fix env flashinfer (#10910)
This commit is contained in:
@@ -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",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user