diff --git a/pkgs/development/python-modules/pyannote-audio/default.nix b/pkgs/development/python-modules/pyannote-audio/default.nix
index 1c407c1e4d3b..8613419900bb 100644
--- a/pkgs/development/python-modules/pyannote-audio/default.nix
+++ b/pkgs/development/python-modules/pyannote-audio/default.nix
@@ -24,6 +24,7 @@
   pyannote-pipeline,
   pyannoteai-sdk,
   pytorch-metric-learning,
+  pyyaml,
   safetensors,
   speechbrain,
   tensorboardx,
@@ -61,11 +62,6 @@ buildPythonPackage (finalAttrs: {
     hatch-vcs
   ];
 
-  pythonRelaxDeps = [
-    "torch"
-    "torchaudio"
-    "torchcodec"
-  ];
   dependencies = [
     asteroid-filterbanks
     einops
@@ -81,6 +77,7 @@ buildPythonPackage (finalAttrs: {
     pyannote-pipeline
     pyannoteai-sdk
     pytorch-metric-learning
+    pyyaml
     safetensors
     speechbrain
     tensorboardx