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

Reworked into separate functions and a commandline, wip

Browse source
This commit is contained in:
MaggotHATE 2023-12-07 17:44:39 +05:00
parent 130aaf836c
commit 549050ad95
2 changed files with 134 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

165
common/common.cpp
View file

@ -128,64 +128,148 @@ nlohmann::json get_json(const char* file_name) noexcept {
return {}; return {};
} }
bool gpt_params_parse(int argc, char ** argv, gpt_params & params) { std::vector<char*> args_parse_json_only(char* file_name) {
bool result = true;
std::vector<char*> arguments_w_json; std::vector<char*> arguments_w_json;
arguments_w_json.push_back(argv[0]); nlohmann::json file_config = get_json(file_name);
int pos = 1; if (!file_config.empty()) { // ensures no unnecessary work
// only the second argument to reduce reading attempts (plus drag'n'drop) arguments_w_json.push_back(file_name);
if (argc > 1) {
// console arguments should override json values, so json processing goes first
nlohmann::json file_config = get_json(argv[1]);
pos = 2; // avoid putting file name into arguments
if (!file_config.empty()) {
for (auto& p : file_config.items()) { for (auto& p : file_config.items()) {
// only use strings, numbers and booleans for switches // only use strings, numbers and booleans for switches
if (p.value().is_string() || p.value().is_number() || p.value().is_boolean()) { if (p.value().is_string() || p.value().is_number() || p.value().is_boolean()) {
char* key = new char[p.key().length() + 1]; char* key = new char[p.key().length() + 1];
strcpy(key, p.key().c_str()); strcpy(key, p.key().c_str());
arguments_w_json.push_back(key); arguments_w_json.push_back(key);
if (!p.value().is_boolean()) { if (!p.value().is_boolean()) {
std::string param_value; std::string param_value;
if (p.value().is_string()) { if (p.value().is_string()) {
param_value = p.value().get<std::string>(); param_value = p.value().get<std::string>();
} else if (p.value().is_number()) { } else if (p.value().is_number()) {
param_value = std::to_string(p.value().get<float>()); // nlohmann::json can't just get numbers as strings, float works fine for int values param_value = std::to_string(p.value().get<float>()); // nlohmann::json can't just get numbers as strings, float works fine for int values
}
char* val = new char[param_value.length() + 1];
strcpy(val, param_value.c_str());
arguments_w_json.push_back(val);
} }
char* val = new char[param_value.length() + 1];
strcpy(val, param_value.c_str());
arguments_w_json.push_back(val);
} }
} }
} }
}
return arguments_w_json;
}
for (int i = pos; i < argc; i++) { std::vector<std::string> args_parse_json_only_string(char* file_name) {
arguments_w_json.push_back(argv[i]); std::vector<std::string> arguments_w_json;
nlohmann::json file_config = get_json(file_name);
if (!file_config.empty()) { // ensures no unnecessary work
arguments_w_json.push_back(file_name);
for (auto& p : file_config.items()) {
// only use strings, numbers and booleans for switches
if (p.value().is_string() || p.value().is_number() || p.value().is_boolean()) {
arguments_w_json.push_back(p.key());
if (!p.value().is_boolean()) {
std::string param_value;
if (p.value().is_string()) {
param_value = p.value().get<std::string>();
} else if (p.value().is_number()) {
param_value = std::to_string(p.value().get<float>()); // nlohmann::json can't just get numbers as strings, float works fine for int values
}
arguments_w_json.push_back(param_value);
}
}
} }
} }
return arguments_w_json;
}
int argc_json = arguments_w_json.size(); // standalone parsing attempt
char** argv_json = &arguments_w_json[0]; bool gpt_params_parse_json(char* file_name, gpt_params & params) {
bool result = true;
//std::vector<const char*> arguments = args_parse_json_only(file_name);
std::vector<std::string> arguments = args_parse_json_only_string(file_name);
if (!arguments.empty()) { // ensures no unnecessary work
int argc_json = arguments.size();
//char** args_json = const_cast<char **>(&arguments[0]);
char** args_json = new char*[arguments.size()];
for(size_t i = 0; i < arguments.size(); i++) {
args_json[i] = new char[arguments[i].size() + 1];
strcpy(args_json[i], arguments[i].c_str());
}
try {
if (!gpt_params_parse_ex(argc_json, args_json, params)) {
gpt_print_usage(argc_json, args_json, gpt_params());
exit(0);
}
for (size_t i = 0; i < arguments.size(); i++) {
delete [] args_json[i];
}
delete [] args_json;
}
catch (const std::invalid_argument & ex) {
fprintf(stderr, "%s\n", ex.what());
gpt_print_usage(argc_json, args_json, gpt_params());
exit(1);
}
} else {
// let's also print help, pointing at a faulty file name/parameter
char** args = new char* {file_name};
gpt_print_usage(1, args, gpt_params());
exit(1);
}
return result;
}
// still deciding which one is worse, both leak
bool gpt_params_parse_json0(char* file_name, gpt_params & params) {
bool result = true;
std::vector<char*> arguments = args_parse_json_only(file_name);
if (!arguments.empty()) { // ensures no unnecessary work
int argc_json = arguments.size();
char** args_json = &arguments[0];
try {
if (!gpt_params_parse_ex(argc_json, args_json, params)) {
gpt_print_usage(argc_json, args_json, gpt_params());
exit(0);
}
}
catch (const std::invalid_argument & ex) {
fprintf(stderr, "%s\n", ex.what());
gpt_print_usage(argc_json, args_json, gpt_params());
exit(1);
}
} else {
// let's also print help, pointing at a faulty file name/parameter
char** args = new char* {file_name};
gpt_print_usage(1, args, gpt_params());
exit(1);
}
return result;
}
bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
bool result = true;
try { try {
if (!gpt_params_parse_ex(argc_json, argv_json, params)) { if (!gpt_params_parse_ex(argc, argv, params)) {
gpt_print_usage(argc_json, argv_json, gpt_params()); gpt_print_usage(argc, argv, gpt_params());
exit(0); exit(0);
} }
} }
catch (const std::invalid_argument & ex) { catch (const std::invalid_argument & ex) {
fprintf(stderr, "%s\n", ex.what()); fprintf(stderr, "%s\n", ex.what());
gpt_print_usage(argc_json, argv_json, gpt_params()); gpt_print_usage(argc, argv, gpt_params());
exit(1); exit(1);
} }
// clearing pointers
for (auto c : arguments_w_json) {
delete [] c;
}
return result; return result;
} }
@ -814,7 +898,10 @@ bool gpt_params_parse_ex(int argc, char ** argv, gpt_params & params) {
// End of Parse args for logging parameters // End of Parse args for logging parameters
#endif // LOG_DISABLE_LOGS #endif // LOG_DISABLE_LOGS
} else { } else {
throw std::invalid_argument("error: unknown argument: " + arg); if (!gpt_params_parse_json0(argv[i], params)) { // attempt to read as a file
invalid_param = true;
throw std::invalid_argument("error: unknown argument: " + arg);
}
} }
} }
if (invalid_param) { if (invalid_param) {

8
common/common.h
View file

@ -138,6 +138,14 @@ struct gpt_params {
nlohmann::json get_json(const char* file_name) noexcept; nlohmann::json get_json(const char* file_name) noexcept;
std::vector<char*> args_parse_json_only(char* file_name);
std::vector<std::string> args_parse_json_only_string(char* file_name);
bool gpt_params_parse_json(char* file_name, gpt_params & params);
bool gpt_params_parse_json0(char* file_name, gpt_params & params);
bool gpt_params_parse_ex(int argc, char ** argv, gpt_params & params); bool gpt_params_parse_ex(int argc, char ** argv, gpt_params & params);
bool gpt_params_parse(int argc, char ** argv, gpt_params & params); bool gpt_params_parse(int argc, char ** argv, gpt_params & params);