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)
|
||||
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)
|
Loading…
Reference in New Issue