vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

slight change to logits

Browse source
This commit is contained in:
Concedo 2023-08-30 21:27:51 +08:00
parent 1301bd7e29
commit a2a4eefa07
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
gpttype_adapter.cpp
View file

@ -369,7 +369,8 @@ static void TokenizeString(const std::string & str_to_tokenize, std::vector<int>
static float LowestLogit(const std::vector<float> & logits) static float LowestLogit(const std::vector<float> & logits)
{ {
int topid = std::min_element(logits.begin(), logits.end()) - logits.begin(); int topid = std::min_element(logits.begin(), logits.end()) - logits.begin();
return (logits[topid] < 0 ? logits[topid] : 0); float v = logits[topid];
return (v < 0 ? (v-1) : 0);
} }
static float LowestLogit(const float *logits, size_t size) static float LowestLogit(const float *logits, size_t size)
{ {
@ -378,7 +379,8 @@ static float LowestLogit(const float *logits, size_t size)
return 0.0; return 0.0;
} }
int topid = std::min_element(logits, logits + size) - logits; int topid = std::min_element(logits, logits + size) - logits;
return (logits[topid] < 0 ? logits[topid] : 0); float v = logits[topid];
return (v < 0 ? (v-1) : 0);
} }
static std::string RemoveBell(const std::string & input) //removes the bell character static std::string RemoveBell(const std::string & input) //removes the bell character