From af45141c62a9b22b8b8bb4b1ea790a0b0a0df073 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=CE=9D=CE=B1=CF=81=CE=BF=CF=85=CF=83=CE=AD=C2=B7=CE=BC?=
 =?UTF-8?q?=C2=B7=CE=B3=CE=B9=CE=BF=CF=85=CE=BC=CE=B5=CE=BC=CE=AF=C2=B7?=
 =?UTF-8?q?=CE=A7=CE=B9=CE=BD=CE=B1=CE=BA=CE=AC=CE=BD=CE=BD=CE=B1?=
 <40709280+NaruseMioShirakana@users.noreply.github.com>
Date: Wed, 24 May 2023 20:45:16 +0800
Subject: [PATCH] Add files via upload

---
 vencoder/HubertSoft_Onnx.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/vencoder/HubertSoft_Onnx.py b/vencoder/HubertSoft_Onnx.py
index 57f37d5..06f10a4 100644
--- 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)
\ No newline at end of file
+        return torch.tensor(logits[0]).transpose(1, 2).to(self.dev)
\ No newline at end of file