update

data_gen/tts/emotion/inference.py

@@ -30,7 +30,7 @@ def load_model(weights_fpath: Path, device=None):
     elif isinstance(device, str):
         _device = torch.device(device)
     _model = EmotionEncoder(_device, torch.device("cpu"))
-    checkpoint = torch.load(weights_fpath)
+    checkpoint = torch.load(weights_fpath, map_location="cpu")
     _model.load_state_dict(checkpoint["model_state"])
     _model.eval()
     print("Loaded encoder trained to step %d" % (checkpoint["step"]))