mirror of https://github.com/coqui-ai/TTS.git
Bug Fix on diffusion inference
This commit is contained in:
parent
b621ab17d1
commit
0664c843d8
|
@ -112,7 +112,7 @@ def load_discrete_vocoder_diffuser(
|
||||||
return SpacedDiffusion(
|
return SpacedDiffusion(
|
||||||
use_timesteps=space_timesteps(trained_diffusion_steps, [desired_diffusion_steps]),
|
use_timesteps=space_timesteps(trained_diffusion_steps, [desired_diffusion_steps]),
|
||||||
model_mean_type="epsilon",
|
model_mean_type="epsilon",
|
||||||
model_vgpt_type="learned_range",
|
model_var_type="learned_range",
|
||||||
loss_type="mse",
|
loss_type="mse",
|
||||||
betas=get_named_beta_schedule("linear", trained_diffusion_steps),
|
betas=get_named_beta_schedule("linear", trained_diffusion_steps),
|
||||||
conditioning_free=cond_free,
|
conditioning_free=cond_free,
|
||||||
|
|
Loading…
Reference in New Issue