This commit is contained in:
mrq 2023-02-05 20:10:40 +00:00
parent daebc6c21c
commit 4ea997106e

View File

@ -170,7 +170,7 @@ def do_spectrogram_diffusion(diffusion_model, diffuser, latents, conditioning_la
noise = torch.randn(output_shape, device=latents.device) * temperature noise = torch.randn(output_shape, device=latents.device) * temperature
diffuser.sampler = sampler.lower() 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}, model_kwargs={'precomputed_aligned_embeddings': precomputed_embeddings},
verbose=verbose, progress=progress, desc=desc) verbose=verbose, progress=progress, desc=desc)