From f66754b5570ed80a79ac48480ded51a973001301 Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 5 Feb 2023 20:10:40 +0000 Subject: [PATCH] oops --- tortoise/api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tortoise/api.py b/tortoise/api.py index b191b54..37a8ec5 100755 --- a/tortoise/api.py +++ b/tortoise/api.py @@ -170,7 +170,7 @@ def do_spectrogram_diffusion(diffusion_model, diffuser, latents, conditioning_la noise = torch.randn(output_shape, device=latents.device) * temperature diffuser.sampler = sampler.lower() - mel = diffuser.ddim_sample_loop(diffusion_model, output_shape, noise=noise, + mel = diffuser.sample_loop(diffusion_model, output_shape, noise=noise, model_kwargs={'precomputed_aligned_embeddings': precomputed_embeddings}, verbose=verbose, progress=progress, desc=desc)