deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mel_head should be optional

Browse Source
This commit is contained in:
James Betker 2022-05-22 12:25:45 -06:00
parent 37640e9759
commit 2dd0b9e6e9
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/music/flat_diffusion.py
View File

@ -271,6 +271,8 @@ class FlatDiffusion(nn.Module):
assert not (return_code_pred and precomputed_code_embeddings is not None), "I cannot compute a code_pred output for you."
unused_params = []
if not return_code_pred:
unused_params.extend(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()))