Fix a bug causing cpu inference unavailable

Author: flutydeer

modules/nsf_hifigan/models.py

@@ -23,7 +23,10 @@ def load_model(model_path, device='cuda'):
 
     generator = Generator(h).to(device)
 
-    cp_dict = torch.load(model_path)
+    if torch.cuda.is_available():
+        cp_dict = torch.load(model_path)
+    else:
+        cp_dict = torch.load(model_path, map_location="cpu")
     generator.load_state_dict(cp_dict['generator'])
     generator.eval()
     generator.remove_weight_norm()