Spaces:
Build error
Build error
Update audio_foundation_models.py
Browse files
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('
|
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 |
|