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

fix unused parameters

Browse Source
This commit is contained in:
James Betker 2022-05-22 08:16:31 -06:00
parent 57d6f6d366
commit 0c60f22197
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -282,8 +282,7 @@ class FlatDiffusion(nn.Module):
else:
code_emb, cond_emb, mel_pred = self.timestep_independent(codes, conditioning_input, x.shape[-1], prenet_latent, True)
if prenet_latent is None:
unused_params.extend(list(self.latent_conditioner.parameters()))
unused_params.extend(list(self.latent_conditioner.parameters()) + [self.latent_fade])
unused_params.append(self.unconditioned_embedding)
blk_emb = self.time_embed(timestep_embedding(timesteps, self.model_channels)) + cond_emb