Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "ca60ad8e55e8c2c43c3b88279fd3351918af8c39"
Unverified Commit 0035f1ce authored by Swipe4057's avatar Swipe4057 Committed by GitHub
Browse files

fix env flashinfer (#10910)

parent 5e21d6ae
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/pyproject_other.toml
View file @ 0035f1ce
...@@ -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