fix(separation): fix division by zero in separation pipeline corner c…

…ases (#1815)
pyannote · Feb 4, 2025 · d6136a8 · d6136a8
1 parent 43bb258
commit d6136a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyannote/audio/pipelines/speech_separation.py b/pyannote/audio/pipelines/speech_separation.py
@@ -656,8 +656,8 @@ def apply(
 
         # separated sources might be scaled up/down due to SI-SDR loss used when training
         # so we peak-normalize them
-        sources.data = sources.data / np.max(
-            np.abs(sources.data), axis=0, keepdims=True
+        sources.data = (
+            sources.data / (np.max(np.abs(sources.data), axis=0, keepdims=True) + 1e-8)
         )
 
         # convert to continuous diarization