From 27f347cdd3a131940ab7d659b4bae03882c1cdb0 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 28 May 2022 23:19:36 -0600 Subject: [PATCH] gotta project that shit! --- 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 428f3254..727fe284 100644 --- a/codes/trainer/eval/music_diffusion_fid.py +++ b/codes/trainer/eval/music_diffusion_fid.py @@ -89,7 +89,7 @@ class MusicDiffusionFid(evaluator.Evaluator): mel = self.spec_fn({'in': audio})['out'] codegen = self.local_modules['codegen'].to(mel.device) - codes = codegen.get_codes(mel) + codes = codegen.get_codes(mel, project=True) mel_norm = normalize_mel(mel) gen_mel = self.diffuser.p_sample_loop(self.model, mel_norm.shape, model_kwargs={'codes': codes, 'conditioning_input': mel_norm[:,:,:140]})