llama : better replace_all (#8852)

This commit is contained in:
Georgi Gerganov 2024-08-05 08:53:39 +03:00 committed by GitHub
parent 064cdc265f
commit f1ea5146d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -122,17 +122,14 @@ static std::string trim(const std::string & str) {
}
static void replace_all(std::string & s, const std::string & search, const std::string & replace) {
std::string result;
for (size_t pos = 0; ; pos += search.length()) {
auto new_pos = s.find(search, pos);
if (new_pos == std::string::npos) {
result += s.substr(pos, s.size() - pos);
break;
if (search.empty()) {
return; // Avoid infinite loop if 'search' is an empty string
}
result += s.substr(pos, new_pos - pos) + replace;
pos = new_pos;
size_t pos = 0;
while ((pos = s.find(search, pos)) != std::string::npos) {
s.replace(pos, search.length(), replace);
pos += replace.length();
}
s = std::move(result);
}
static bool is_float_close(float a, float b, float abs_tol) {