From 798166015a0c6609f9d15aac19f41d7b8354dac4 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 12 Jun 2022 14:51:56 -0600 Subject: [PATCH] provide conditioning ijnput as mel_norm --- codes/trainer/eval/music_diffusion_fid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/trainer/eval/music_diffusion_fid.py b/codes/trainer/eval/music_diffusion_fid.py index 415f48b0..25a12168 100644 --- a/codes/trainer/eval/music_diffusion_fid.py +++ b/codes/trainer/eval/music_diffusion_fid.py @@ -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)