Skip to content

Replace logits_to_logprobs implementation with numpy equivalent to llama.cpp#991

Merged
abetlen merged 1 commit intoabetlen:mainfrom
player1537:patch-1
Dec 12, 2023
Merged

Replace logits_to_logprobs implementation with numpy equivalent to llama.cpp#991
abetlen merged 1 commit intoabetlen:mainfrom
player1537:patch-1

Conversation

@player1537
Copy link
Copy Markdown
Contributor

@player1537 player1537 commented Dec 9, 2023

See #990. This change makes the logits_to_logprobs function equivalent to the version in the llama.cpp repository. It uses numpy so it's much faster than the previous version.

Edit: I should add that I'm not 100% confident in this change. I'd appreciate if someone else could test this themselves too.

…ama.cpp

See abetlen#990. This change makes the logits_to_logprobs function equivalent to the version in the llama.cpp repository. It uses numpy so it's much faster than the previous version.
@abetlen abetlen merged commit ef22e47 into abetlen:main Dec 12, 2023
@abetlen
Copy link
Copy Markdown
Owner

abetlen commented Dec 12, 2023

@player1537 perfect, thank you very much!

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.

2 participants