Add files via upload
This commit is contained in:
parent
bf31717403
commit
af45141c62
|
@ -22,7 +22,7 @@ class HubertSoft_Onnx(SpeechEncoder):
|
||||||
feats = feats.mean(-1)
|
feats = feats.mean(-1)
|
||||||
assert feats.dim() == 1, feats.dim()
|
assert feats.dim() == 1, feats.dim()
|
||||||
feats = feats.view(1, -1)
|
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}
|
onnx_input = {self.model.get_inputs()[0].name: feats}
|
||||||
logits = self.model.run(None, onnx_input)
|
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)
|
Loading…
Reference in New Issue