Update Cutedsl version and pin cuda-python version (#16838)
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user