From 0b425c9bd1f34a919eb0ecfc4e17bf322aa9fe9d Mon Sep 17 00:00:00 2001 From: "Salvador E. Tropea" Date: Fri, 18 Jul 2025 11:03:54 -0300 Subject: [PATCH] [Demucs][Removed] Normalization Not really needed --- source/inference/demixer.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/source/inference/demixer.py b/source/inference/demixer.py index 47d639c..87ce06f 100644 --- a/source/inference/demixer.py +++ b/source/inference/demixer.py @@ -254,12 +254,6 @@ class DemixerDemucs(DemixerGeneric): else: logger.debug(f"Using user provided segment size {forced_segment} s") - # Normalize the input tensors - ref = input_tensor_on_device.mean(1, keepdim=True) - mean = ref.mean(dim=2, keepdim=True) - std = ref.std(dim=2, keepdim=True) - input_tensor_on_device = (input_tensor_on_device - mean) / (std + 1e-8) - if self.d.get('use_demucs_pt_process', False): # This is for the model from torchaudio, using the Demucs code we get some strange noises # Using their example they aren't produced @@ -292,11 +286,6 @@ class DemixerDemucs(DemixerGeneric): callback=lambda x: (x.get('state') == 'end') and comfy_progress_bar.update(1) if with_comfy else None, ) - # Denormalize the output stems - mean = mean.unsqueeze(1) - std = std.unsqueeze(1) - separated_tensors = separated_tensors * std + mean - # Move the final result tensor back to the CPU before creating the output dicts. # This is good practice to free up VRAM for subsequent nodes. separated_tensors = separated_tensors.cpu()