Update Cutedsl version and pin cuda-python version (#16838)

This commit is contained in:
Baizhou Zhang
2026-01-10 17:08:43 +08:00
committed by GitHub
parent 3c358736d1
commit 9fd2358cc2
2 changed files with 3 additions and 4 deletions

View File

@@ -22,7 +22,7 @@ dependencies = [
"blobfile==3.0.0",
"build",
"compressed-tensors",
"cuda-python",
"cuda-python==12.9",
"decord2",
"datasets",
"einops",
@@ -38,7 +38,7 @@ dependencies = [
"msgspec",
"ninja",
"numpy",
"nvidia-cutlass-dsl==4.2.1",
"nvidia-cutlass-dsl==4.3.1",
"nvidia-ml-py",
"openai-harmony==0.0.4",
"openai==2.6.1",