diff --git a/pyannote/audio/pipelines/speech_separation.py b/pyannote/audio/pipelines/speech_separation.py index 55f4ef5e7..c5ecf6dd8 100644 --- 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