Bug Fix on diffusion inference

coqui-ai · Nov 3, 2023 · fbc5e2a · fbc5e2a
1 parent ac7c008
commit fbc5e2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,