fix CPU only error

converter.py

@@ -20,6 +20,7 @@ from torch.nn.utils import weight_norm, remove_weight_norm, spectral_norm
 
 MAX_WAV_VALUE = 32768.0
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def load_wav(full_path):
     sampling_rate, data = read(full_path)
@@ -352,7 +353,8 @@ class Generator(torch.nn.Module):
         config = get_config(config_path)
         vocoder = cls(config)
 
-        state_dict_g = torch.load(ckpt_path)
+        
+        state_dict_g = torch.load(ckpt_path, map_location=device)
         vocoder.load_state_dict(state_dict_g["generator"])
         vocoder.eval()
         vocoder.remove_weight_norm()

requirements.txt

@@ -6,4 +6,5 @@ matplotlib
 numpy
 scipy
 Pillow
-librosa
+librosa
+transformers