Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use PyTorch as logits transpose for ONNX support (#141)

Browse Source
This commit is contained in:
Michael Goin 2022-09-26 13:54:26 -04:00 committed by GitHub
parent 2037b65f3f
commit 9c8183a179
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/model.py
View File

@ -189,7 +189,7 @@ class TextDecoder(nn.Module):
x = block(x, xa, mask=self.mask, kv_cache=kv_cache)
x = self.ln(x)
logits = (x @ self.token_embedding.weight.to(x.dtype).T).float()
logits = (x @ torch.transpose(self.token_embedding.weight.to(x.dtype), 0, 1)).float()
return logits