Skip to content

Commit 90eac14

Browse files
[AudioLDM] Fix dtype of returned waveform (huggingface#3189)
1 parent 11f527a commit 90eac14

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/audioldm/pipeline_audioldm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ def mel_spectrogram_to_waveform(self, mel_spectrogram):
293293

294294
waveform = self.vocoder(mel_spectrogram)
295295
# we always cast to float32 as this does not cause significant overhead and is compatible with bfloat16
296-
waveform = waveform.cpu()
296+
waveform = waveform.cpu().float()
297297
return waveform
298298

299299
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_extra_step_kwargs

0 commit comments

Comments
 (0)