lmzjms commited on
Commit
405937f
1 Parent(s): da9b785

Update audio_foundation_models.py

Browse files
Files changed (1) hide show
  1. audio_foundation_models.py +1 -1
audio_foundation_models.py CHANGED
@@ -292,7 +292,7 @@ class Inpaint:
292
  print("Initializing Make-An-Audio-inpaint to %s" % device)
293
  self.device = device
294
  self.sampler = initialize_model_inpaint('text_to_audio/Make_An_Audio/configs/inpaint/txt2audio_args.yaml', 'text_to_audio/Make_An_Audio/useful_ckpts/inpaint7_epoch00047.ckpt')
295
- self.vocoder = VocoderBigVGAN('./vocoder/logs/bigv16k53w',device=device)
296
  self.cmap_transform = matplotlib.cm.viridis
297
  def make_batch_sd(self, mel, mask, num_samples=1):
298
 
 
292
  print("Initializing Make-An-Audio-inpaint to %s" % device)
293
  self.device = device
294
  self.sampler = initialize_model_inpaint('text_to_audio/Make_An_Audio/configs/inpaint/txt2audio_args.yaml', 'text_to_audio/Make_An_Audio/useful_ckpts/inpaint7_epoch00047.ckpt')
295
+ self.vocoder = VocoderBigVGAN('text_to_audio/Make_An_Audio/vocoder/logs/bigv16k53w',device=device)
296
  self.cmap_transform = matplotlib.cm.viridis
297
  def make_batch_sd(self, mel, mask, num_samples=1):
298