wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-16 03:57:27 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Merge ddac8fd48e into 7a063e83a7

Browse Source
This commit is contained in:
Jean J. de Jong 2026-05-14 21:41:11 +02:00 committed by GitHub
commit 89c45d584f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/lightricks/vae/audio_vae.py
View File

@ -154,7 +154,7 @@ class AudioVAE(torch.nn.Module):
waveform, waveform_sample_rate, device=waveform.device
)
latents = self.autoencoder.encode(mel_spec)
latents = self.autoencoder.encode(mel_spec.to(dtype=waveform.dtype))
posterior = DiagonalGaussianDistribution(latents)
latent_mode = posterior.mode()