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 |
|