Skip to content

GitLab

Unverified Commit 90eac14f authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

[AudioLDM] Fix dtype of returned waveform (#3189)

parent 11f527ac
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/audioldm/pipeline_audioldm.py
View file @ 90eac14f
...@@ -293,7 +293,7 @@ class AudioLDMPipeline(DiffusionPipeline): ...@@ -293,7 +293,7 @@ class AudioLDMPipeline(DiffusionPipeline):
waveform = self.vocoder(mel_spectrogram) waveform = self.vocoder(mel_spectrogram)
# we always cast to float32 as this does not cause significant overhead and is compatible with bfloat16 # we always cast to float32 as this does not cause significant overhead and is compatible with bfloat16
waveform = waveform.cpu() waveform = waveform.cpu().float()
return waveform return waveform
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_extra_step_kwargs # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_extra_step_kwargs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment