Add files via upload

2023-05-24 20:45:16 +08:00 · 2023-05-24 20:45:16 +08:00 · af45141c62
parent bf31717403
commit af45141c62
1 changed files with 2 additions and 2 deletions
--- a/vencoder/HubertSoft_Onnx.py
+++ b/vencoder/HubertSoft_Onnx.py
@ -22,7 +22,7 @@ class HubertSoft_Onnx(SpeechEncoder):
          feats = feats.mean(-1)
        assert feats.dim() == 1, feats.dim()
        feats = feats.view(1, -1)
-        feats = feats.unsqueeze(0).detach().numpy()
+        feats = feats.unsqueeze(0).cpu().detach().numpy()
        onnx_input = {self.model.get_inputs()[0].name: feats}
        logits = self.model.run(None, onnx_input)
-        return torch.tensor(logits[0]).transpose(1, 2)
+        return torch.tensor(logits[0]).transpose(1, 2).to(self.dev)