mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-14 19:17:32 +08:00
Merge ddac8fd48e into 26515acd23
This commit is contained in:
commit
8f26ccc20d
@ -154,7 +154,7 @@ class AudioVAE(torch.nn.Module):
|
|||||||
waveform, waveform_sample_rate, device=waveform.device
|
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)
|
posterior = DiagonalGaussianDistribution(latents)
|
||||||
latent_mode = posterior.mode()
|
latent_mode = posterior.mode()
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user