sampling : deduplicated code for probability distribution access (#6240)

Browse source

* sampling: remove duplicated code for probability distribution access

* free original_logits

* fix original_logits allocation

* fixes based on review @cebtenzzre

* change function name to `llama_sampling_prepare`

...

This commit is contained in:

Minsoo Cheong 2024-03-24 17:54:07 +09:00 • committed by GitHub

parent ddf6568510

commit 586e7bc561

No known key found for this signature in database

GPG key ID: B5690EEEBB952194

4 changed files with 28 additions and 76 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download patch file Download diff file Expand all files Collapse all files

BIN

retrieval Executable file

View file

Binary file not shown.