Spaces:

patriotyk
/

pflowtts_ukr_demo

App Files Files Community

Serhiy Stetskovych commited on May 11

Commit

5db46b8

•

1 Parent(s): 2dd65dd

Fix model load.

Files changed (1) hide show

app.py +2 -2

app.py CHANGED Viewed

@@ -116,9 +116,9 @@ _ = model.eval()
 hifigan = load_hifigan(HIFIGAN_MODEL_PATH, device)
-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, device_map=device)
 denoiser = None#Denoiser(vocoder, mode="zeros")

 hifigan = load_hifigan(HIFIGAN_MODEL_PATH, device)
+vocos_22050 = Vocos.from_pretrained(VOCODER22_MODEL_PATH).to(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).to(device)
 denoiser = None#Denoiser(vocoder, mode="zeros")