From 73d7211a4cbb161d44cb0362cd8e15d90f09070a Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 19 Jul 2022 11:17:43 -0600 Subject: [PATCH] fix script --- codes/models/audio/music/tfdpc_v5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/tfdpc_v5.py b/codes/models/audio/music/tfdpc_v5.py index bbe5ee50..a71650ae 100644 --- a/codes/models/audio/music/tfdpc_v5.py +++ b/codes/models/audio/music/tfdpc_v5.py @@ -432,7 +432,7 @@ def inference_tfdpc5_with_cheater(): spectral_diffuser = SpacedDiffusion(use_timesteps=space_timesteps(4000, [32]), model_mean_type='epsilon', model_var_type='learned_range', loss_type='mse', betas=get_named_beta_schedule('linear', 4000), conditioning_free=True, conditioning_free_k=1) - from trainer.injectors.audio_injectors import + from trainer.injectors.audio_injectors import denormalize_torch_mel gen_mel_denorm = denormalize_torch_mel(gen_mel) output_shape = (1,16,gen_mel_denorm.shape[-1]*256//16) gen_wav = spectral_diffuser.ddim_sample_loop(m2w, output_shape, model_kwargs={'codes': gen_mel_denorm})