Update utils.h
This commit is contained in:
parent
2913d75a42
commit
90b9b5b8c0
1 changed files with 1 additions and 1 deletions
2
utils.h
2
utils.h
|
@ -88,7 +88,7 @@ gpt_vocab::id llama_sample_top_p_top_k(
|
||||||
std::mt19937 & rng);
|
std::mt19937 & rng);
|
||||||
|
|
||||||
// filer to top K tokens from list of logits
|
// filer to top K tokens from list of logits
|
||||||
void sample_top_k(std::vector<std::pair<double, gpt_vocab::id>> &logits_id, int top_k);
|
void sample_top_k(std::vector<std::pair<double, gpt_vocab::id>> & logits_id, int top_k);
|
||||||
|
|
||||||
//
|
//
|
||||||
// Quantization
|
// Quantization
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue