chore: rm nvidia deps

2026-04-30 00:07:42 +03:00
parent 4cb8c317d6
commit 4b6a5e703b
3 changed files with 108 additions and 205 deletions
@@ -22,6 +22,7 @@ dependencies = [
  "pydub>=0.25.1",
  "google-cloud-speech>=2.34.0",
  "openai-whisper>=20250625",
+  "torch; sys_platform == 'linux' and platform_machine == 'x86_64'",
  "dramatiq[redis]>=1.17.0",
  "redis>=5.0.0",
  "psycopg2-binary>=2.9.9",
@@ -44,6 +45,16 @@ tools = [
  "radon",
 ]

+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
 [tool.pytest.ini_options]
 asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "function"