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 authored and GitHub committed 2024-03-24 10:54:07 +02:00

1 parent ddf6568510

commit 586e7bc561

4 files changed

+28 -76

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

No files matched your search

retrieval

Executable

BIN

Binary file not shown.