From 3cf78e3c44bfb550b642684f835d4b72020d1846 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 10 Oct 2022 12:10:56 -0600 Subject: [PATCH] train mel head even when not --- codes/models/audio/tts/unet_diffusion_tts_flat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/tts/unet_diffusion_tts_flat.py b/codes/models/audio/tts/unet_diffusion_tts_flat.py index 2280d5ba..f7db26ab 100644 --- a/codes/models/audio/tts/unet_diffusion_tts_flat.py +++ b/codes/models/audio/tts/unet_diffusion_tts_flat.py @@ -304,7 +304,7 @@ class DiffusionTtsFlat(nn.Module): assert precomputed_aligned_embeddings is not None or (aligned_conditioning is not None and conditioning_input is not None) assert not (return_code_pred and precomputed_aligned_embeddings is not None) # These two are mutually exclusive. - unused_params = [] + unused_params = [self.mel_head.parameters()] if conditioning_free: code_emb = self.unconditioned_embedding.repeat(x.shape[0], 1, x.shape[-1]) unused_params.extend(list(self.code_converter.parameters()) + list(self.code_embedding.parameters()))