Update Torch and Torchaudio dependencies for multi-architecture support

CHANGELOG.md

 Changelog
 =========
 
-[1.8.1] (2025-02-17)
+[1.8.1] (2025-02-18)
 --------------------
 
 ### Fixed
 
 - Fixed issues with Torch CUDA and cuDNN
+- Updated Torch and Torchaudio dependencies for multi-architecture support
 
 [1.8.0] (2025-02-17)
 --------------------

pyproject.toml

@@ -29,8 +29,16 @@ whisperx = "^3.3.1"
 tqdm = "^4.67.1"
 llvmlite = "^0.44.0"
 numba = "^0.61.0"
-torch = { version = "2.6.0", source = "pytorch-cpu" }
-torchaudio = { version = "2.6.0", source = "pytorch-cpu" }
+torch = [
+    { version = "2.6.0", source = "pypi", markers = "sys_platform == 'darwin'"},
+    { version = "2.6.0", source = "pypi", markers = "platform_machine == 'aarch64' and sys_platform != 'darwin'"},
+    { version = "2.6.0", source = "pytorch-cpu", markers = "platform_machine == 'x86_64' and sys_platform != 'darwin'" },
+]
+torchaudio = [
+    { version = "2.6.0", source = "pypi", markers = "sys_platform == 'darwin'"},
+    { version = "2.6.0", source = "pypi", markers = "platform_machine == 'aarch64' and sys_platform != 'darwin'"},
+    { version = "2.6.0", source = "pytorch-cpu", markers = "platform_machine == 'x86_64' and sys_platform != 'darwin'" },
+]
 
 [poetry.group.dev.dependencies]
 pytest = "^8.3.4"