Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-19 01:05:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

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
gpttype_adapter.cpp
+4 -2
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