diff --git a/fastvideo-kernel/pyproject.toml b/fastvideo-kernel/pyproject.toml index b9a7a73fb..c5460e0c5 100644 --- a/fastvideo-kernel/pyproject.toml +++ b/fastvideo-kernel/pyproject.toml @@ -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'", ]