provide conditioning ijnput as mel_norm

This commit is contained in:
James Betker 2022-06-12 14:51:56 -06:00
parent 0c95be1624
commit 798166015a

View File

@ -131,7 +131,7 @@ class MusicDiffusionFid(evaluator.Evaluator):
# return x
gen_mel = self.diffuser.p_sample_loop(self.model, mel_norm.shape, #denoised_fn=denoising_fn, clip_denoised=False,
model_kwargs={'truth_mel': mel_norm,
'conditioning_input': None,
'conditioning_input': mel_norm,
'disable_diversity': True})
gen_mel_denorm = denormalize_mel(gen_mel)