kevinlee819

Thanks for your reply! If I don't need all the logits for each token, but just the topk logits for a certain token and the token corresponding to those logits,...

Thank you for your reply, if I want to modify the source code to achieve my goal, what files do I need to modify, can you give me some advice?...