From 9502e0755e171e39dce651c07ca1a9a8797a0e8d Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 10 Oct 2022 12:15:51 -0600 Subject: [PATCH] ugh --- 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 4b008743..6b034680 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 = [list(self.mel_head.parameters())] + unused_params = list(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()))