[Application] Bugfix in LLaMA Application

nnstreamer / nntrainer

NNtrainer is Software Framework for Training Neural Network Models on Devices.

Apache License 2.0

135 stars 71 forks source link

[Application] Bugfix in LLaMA Application #2553

Closed EunjuYang closed 2 months ago

EunjuYang commented 2 months ago

This commit fixes a bug in applyTKP function.
It seems applying Top-K and Top-P to logits didn't work as intended

Self evaluation:

Build test: [X]Passed [ ]Failed [ ]Skipped Run test: [X]Passed [ ]Failed [ ]Skipped

taos-ci commented 2 months ago

:memo: TAOS-CI Version: 1.5.20200925. Thank you for submitting PR #2553. Please a submit 1commit/1PR (one commit per one PR) policy to get comments quickly from reviewers. Your PR must pass all verificiation processes of cibot before starting a review process from reviewers. If you are new member to join this project, please read manuals in documentation folder and wiki page. In order to monitor a progress status of your PR in more detail, visit http://ci.nnstreamer.ai/.