Fix model load.

.gitattributes

@@ -39,3 +39,4 @@ checkpoint_epoch=499.ckpt filter=lfs diff=lfs merge=lfs -text
 checkpoint_epoch=599.ckpt filter=lfs diff=lfs merge=lfs -text
 checkpoint_epoch=649.ckpt filter=lfs diff=lfs merge=lfs -text
 g_00140000_m filter=lfs diff=lfs merge=lfs -text
+checkpoints/g_00120000 filter=lfs diff=lfs merge=lfs -text

app.py

@@ -116,9 +116,9 @@ _ = model.eval()
 
 
 hifigan = load_hifigan(HIFIGAN_MODEL_PATH, device)
-vocos_22050 = Vocos.from_pretrained(VOCODER22_MODEL_PATH)
+vocos_22050 = Vocos.from_pretrained(VOCODER22_MODEL_PATH, device_map=device)
 #vocos_44100 = load_vocos('checkpoints/vocos_checkpoint_epoch=209_step=3924480_val_loss=3.7036_44100_11.ckpt', 'vocos.yaml', device)
-vocos_44100 = Vocos.from_pretrained(VOCODER44_MODEL_PATH)
+vocos_44100 = Vocos.from_pretrained(VOCODER44_MODEL_PATH, device_map=device)
 denoiser = None#Denoiser(vocoder, mode="zeros")