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,