mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

and listify them

Browse Source
This commit is contained in:
James Betker 2022-10-10 12:13:49 -06:00
parent 3cf78e3c44
commit fce2c8f5db
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/tts/unet_diffusion_tts_flat.py
View File

@ -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()))