Skip to content

fix issue with return_logits_vocab option#2020

Draft
cmullovisl wants to merge 1 commit intoOpenNMT:masterfrom
cmullovisl:master
Draft

fix issue with return_logits_vocab option#2020
cmullovisl wants to merge 1 commit intoOpenNMT:masterfrom
cmullovisl:master

Conversation

@cmullovisl
Copy link

When setting the return_logits_vocab option, the logits of type StorageView are saved with emplace_back in the DecodingResult.logits_vocab of type std::vector<std::vector<StorageView>>. The type mismatch (StorageView instead of std::vector<StorageView>) results in an empty StorageView being emplaced intead of the computed logits.

@cmullovisl cmullovisl marked this pull request as draft February 27, 2026 16:51
@cmullovisl
Copy link
Author

Sorry, the current draft only saves the logits of one of the tokens. I will edit it to save all the necessary logits.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant