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

eval bug fix

Browse Source
This commit is contained in:
James Betker 2022-06-10 13:51:06 -06:00
parent 84469f3538
commit 89bd40d39f
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/transformer_diffusion8.py
View File

@ -234,6 +234,8 @@ class TransformerDiffusionWithQuantizer(nn.Module):
diff = self.diff(x, timesteps, codes=proj, conditioning_input=conditioning_input, diff = self.diff(x, timesteps, codes=proj, conditioning_input=conditioning_input,
conditioning_free=conditioning_free) conditioning_free=conditioning_free)
if disable_diversity:
return diff
if mse is None: if mse is None:
return diff, diversity_loss return diff, diversity_loss
return diff, diversity_loss, mse return diff, diversity_loss, mse