Open SaeedNajafi opened 1 month ago
Don't you have a bug in the logit processor? It seems you don't even use the processed scores in the wrapper.
https://github.com/microsoft/LMOps/blob/96e7d9d086cba6cd878c98ac810902eae1323d8b/minillm/transformers/src/transformers/generation/utils.py#L2942
Thanks for pointing it out! We have fixed this issue in this commit.
Don't you have a bug in the logit processor? It seems you don't even use the processed scores in the wrapper.
https://github.com/microsoft/LMOps/blob/96e7d9d086cba6cd878c98ac810902eae1323d8b/minillm/transformers/src/transformers/generation/utils.py#L2942