From b203a7dc9716e8908504be168464f4845a1de014 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 19 Jul 2022 15:05:12 -0600 Subject: [PATCH] And remove unused parameters --- codes/trainer/eval/music_diffusion_fid.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/codes/trainer/eval/music_diffusion_fid.py b/codes/trainer/eval/music_diffusion_fid.py index 7a431b2e..af79598e 100644 --- a/codes/trainer/eval/music_diffusion_fid.py +++ b/codes/trainer/eval/music_diffusion_fid.py @@ -151,9 +151,7 @@ class MusicDiffusionFid(evaluator.Evaluator): # x = x.clamp(-s, s) / s # 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': mel_norm, - 'disable_diversity': True}) + model_kwargs={'truth_mel': mel_norm}) gen_mel_denorm = denormalize_mel(gen_mel) output_shape = (1,16,audio.shape[-1]//16)