From fce2c8f5db4fb26dad28e45a4e05b413634872ba Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 10 Oct 2022 12:13:49 -0600 Subject: [PATCH] and listify them --- 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 f7db26ab..4b008743 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 = [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()))