Hugging Face's logo

Spaces:
patriotyk
/
pflowtts_ukr_demo
Running on Zero

App Files Community
2
Serhiy Stetskovych commited on
Commit
5db46b8
·
1 Parent(s): 2dd65dd

Fix model load.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -116,9 +116,9 @@ _ = model.eval()
116
 
117
 
118
  hifigan = load_hifigan(HIFIGAN_MODEL_PATH, device)
119
- vocos_22050 = Vocos.from_pretrained(VOCODER22_MODEL_PATH, device_map=device)
120
  #vocos_44100 = load_vocos('checkpoints/vocos_checkpoint_epoch=209_step=3924480_val_loss=3.7036_44100_11.ckpt', 'vocos.yaml', device)
121
- vocos_44100 = Vocos.from_pretrained(VOCODER44_MODEL_PATH, device_map=device)
122
  denoiser = None#Denoiser(vocoder, mode="zeros")
123
 
124
 
 
116
 
117
 
118
  hifigan = load_hifigan(HIFIGAN_MODEL_PATH, device)
119
+ vocos_22050 = Vocos.from_pretrained(VOCODER22_MODEL_PATH).to(device)
120
  #vocos_44100 = load_vocos('checkpoints/vocos_checkpoint_epoch=209_step=3924480_val_loss=3.7036_44100_11.ckpt', 'vocos.yaml', device)
121
+ vocos_44100 = Vocos.from_pretrained(VOCODER44_MODEL_PATH).to(device)
122
  denoiser = None#Denoiser(vocoder, mode="zeros")
123
 
124