jacob-c commited on
Commit
26979eb
·
1 Parent(s): f69d840
audioldm/variational_autoencoder/autoencoder.py CHANGED
@@ -7,7 +7,7 @@ from audioldm.variational_autoencoder.distributions import DiagonalGaussianDistr
7
  from audioldm.bigvgan import BigVGANVocoder
8
 
9
  # After you create your latent_diffusion instance:
10
- latent_diffusion.vocoder = BigVGANVocoder(device='cuda', use_cuda_kernel=False)
11
 
12
 
13
  class AutoencoderKL(nn.Module):
 
7
  from audioldm.bigvgan import BigVGANVocoder
8
 
9
  # After you create your latent_diffusion instance:
10
+ #latent_diffusion.vocoder = BigVGANVocoder(device='cuda', use_cuda_kernel=False)
11
 
12
 
13
  class AutoencoderKL(nn.Module):