From fbc5e2a0c2e9751c76c82643c88219d704353c19 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Fri, 3 Nov 2023 14:12:07 -0300 Subject: [PATCH] Bug Fix on diffusion inference --- TTS/tts/models/xtts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/models/xtts.py b/TTS/tts/models/xtts.py index eac91e34f8..bb5c85b0a8 100644 --- a/TTS/tts/models/xtts.py +++ b/TTS/tts/models/xtts.py @@ -112,7 +112,7 @@ def load_discrete_vocoder_diffuser( return SpacedDiffusion( use_timesteps=space_timesteps(trained_diffusion_steps, [desired_diffusion_steps]), model_mean_type="epsilon", - model_vgpt_type="learned_range", + model_var_type="learned_range", loss_type="mse", betas=get_named_beta_schedule("linear", trained_diffusion_steps), conditioning_free=cond_free,