Update inference/m4singer/base_svs_infer.py

inference/m4singer/base_svs_infer.py

@@ -18,7 +18,7 @@ import re
 class BaseSVSInfer:
     def __init__(self, hparams, device=None):
         if device is None:
-            device = 'cuda'   #if torch.cuda.is_available() else 'cpu'
+            device = 'cuda' if torch.cuda.is_available() else 'cpu'
         self.hparams = hparams
         self.device = device
 
@@ -51,7 +51,7 @@ class BaseSVSInfer:
         ckpt = sorted(glob.glob(f'{base_dir}/model_ckpt_steps_*.ckpt'), key=
         lambda x: int(re.findall(f'{base_dir}/model_ckpt_steps_(\d+).ckpt', x)[0]))[-1]
         print('| load HifiGAN: ', ckpt)
-        ckpt_dict = torch.load(ckpt)   #torch.load(ckpt, map_location="cpu")
+        ckpt_dict = torch.load(ckpt, map_location="cpu")
         config = set_hparams(config_path, global_hparams=False)
         state = ckpt_dict["state_dict"]["model_gen"]
         vocoder = HifiGanGenerator(config)