Bug Fix on diffusion inference

This commit is contained in:
Edresson Casanova 2023-11-03 14:12:07 -03:00 committed by Eren G??lge
parent b621ab17d1
commit 0664c843d8
1 changed files with 1 additions and 1 deletions

View File

@ -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,