fix: lock PyTorch CUDA builds for transcription

2026-07-14 15:30:53 +08:00
parent 790c775eac
commit 23e0529461
4 changed files with 98 additions and 18 deletions
@@ -17,6 +17,19 @@ funasr-nano = "funasr_nano:main"
 funasr-fast = "funasr_fast:main"
 funasr-regular = "funasr_regular:main"

+[[tool.uv.index]]
+name = "pytorch-cu130"
+url = "https://download.pytorch.org/whl/cu130"
+explicit = true
+
+[tool.uv.sources]
+torch = [
+    { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+torchaudio = [
+    { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+
 [build-system]
 requires = ["setuptools>=69"]
 build-backend = "setuptools.build_meta"