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

make code pred returns optional

Browse Source
This commit is contained in:
James Betker 2022-03-26 08:33:30 -06:00
parent 2a29a71c37
commit 6909f196b4
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/audio/tts/unet_diffusion_tts_flat0.py
View File

@ -187,7 +187,7 @@ class DiffusionTtsFlat(nn.Module):
} }
return groups return groups
def forward(self, x, timesteps, aligned_conditioning, conditioning_input, conditioning_free=False): def forward(self, x, timesteps, aligned_conditioning, conditioning_input, conditioning_free=False, return_code_pred=False):
""" """
Apply the model to an input batch. Apply the model to an input batch.
@ -256,7 +256,9 @@ class DiffusionTtsFlat(nn.Module):
extraneous_addition = extraneous_addition + p.mean() extraneous_addition = extraneous_addition + p.mean()
out = out + extraneous_addition * 0 out = out + extraneous_addition * 0
if return_code_pred:
return out, mel_pred return out, mel_pred
return out
@register_model @register_model