From d2fd63e7a6d82bf9fd721f978ea45b77d7b3c643 Mon Sep 17 00:00:00 2001 From: KyleZhang1118 Date: Wed, 6 Nov 2024 21:38:31 +0800 Subject: [PATCH] fix format --- wesep/cli/extractor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wesep/cli/extractor.py b/wesep/cli/extractor.py index 3827660..b14f0ba 100644 --- a/wesep/cli/extractor.py +++ b/wesep/cli/extractor.py @@ -150,7 +150,7 @@ def extract_speech_from_pcm(self, target_speech = outputs.to(torch.device("cpu")) if self.output_norm: target_speech = ( - target_speech + target_speech / abs(target_speech).max(dim=1, keepdim=True).values * 0.9 ) return target_speech