fix vocos load

app.py

@@ -75,9 +75,9 @@ def load_hifigan(checkpoint_path, device):
 
 
 def load_vocos(checkpoint_path, config_path, device):
-    model = Vocos.from_hparams(config_path)
+    model = Vocos.from_hparams(config_path).to(device)
 
-    raw_model = torch.load(checkpoint_path, map_location=torch.device('cpu'))
+    raw_model = torch.load(checkpoint_path, map_location=device)
     raw_model = raw_model if 'state_dict' not in raw_model else raw_model['state_dict']
     model.load_state_dict(raw_model, strict=False)
     model.eval()
@@ -137,7 +137,7 @@ def synthesise(text, speed):
         guidance_scale=1.5
 
     )
-    waveform_vocos = vocos.decode(output["mel"]).to(device).squeeze()
+    waveform_vocos = vocos.decode(output["mel"]).cpu().squeeze()
     #waveform_vocos_44100 = vocos_44100.decode(output["mel"]).cpu().squeeze()
     #waveform_hifigan = hifigan(output["mel"]).clamp(-1, 1).cpu().squeeze()
     #transform = torchaudio.transforms.Vol(gain=-18, gain_type="db")