wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-11 08:57:22 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Merge ad3797ad4f into 091b70edda

Browse Source
This commit is contained in:
Squishy Code AI 2026-07-08 23:14:12 +08:00 committed by GitHub
commit 057656f8b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -154,6 +154,10 @@ class AudioVAE(torch.nn.Module):
waveform, waveform_sample_rate, device=waveform.device waveform, waveform_sample_rate, device=waveform.device
) )
# Cast mel spectrogram to match encoder weight dtype (bf16 VAE + float32 audio input)
encoder_dtype = next(self.autoencoder.encoder.parameters()).dtype
mel_spec = mel_spec.to(dtype=encoder_dtype)
latents = self.autoencoder.encode(mel_spec) latents = self.autoencoder.encode(mel_spec)
posterior = DiagonalGaussianDistribution(latents) posterior = DiagonalGaussianDistribution(latents)
latent_mode = posterior.mode() latent_mode = posterior.mode()