[bugfix]: pin fastvideo-kernel to Torch 2.12.0 (#1774)
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
[build-system]
|
||||
requires = [
|
||||
"scikit-build-core>=0.10",
|
||||
"torch>=2.5.0",
|
||||
"torch==2.12.0",
|
||||
"setuptools>=61.0.0",
|
||||
"wheel"
|
||||
]
|
||||
@@ -22,7 +22,7 @@ classifiers = [
|
||||
"Environment :: GPU :: NVIDIA CUDA",
|
||||
]
|
||||
dependencies = [
|
||||
"torch>=2.5.0",
|
||||
"torch==2.12.0",
|
||||
"triton>=2.0.0; sys_platform == 'linux'",
|
||||
]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user