And that causes all values in self.token_bias to be set to -inf (I don't know Python/Pytorch well enough to know why that is).
Please close this if there's a more appropriate change/refactor to fix this, assuming it is actually a bug. Of if you'd just rather do it on your machine since it's such a simple change. I guess you might also want to add a check in disallow_tokens to catch future bugs like this.
I might be misunderstanding something here, but this code in
websocket_actions.py
:Combined with this code that follows:
Means that we may be passing
None
todisallow_tokens
, which looks like this:And that causes all values in
self.token_bias
to be set to-inf
(I don't know Python/Pytorch well enough to know why that is).Please close this if there's a more appropriate change/refactor to fix this, assuming it is actually a bug. Of if you'd just rather do it on your machine since it's such a simple change. I guess you might also want to add a check in
disallow_tokens
to catch future bugs like this.Thanks!