This commit is contained in:
Xuan Son Nguyen 2024-12-08 21:12:18 +01:00
parent 6ec3f77a41
commit a4d2572494

View file

@ -3519,7 +3519,7 @@ int main(int argc, char ** argv) {
std::string prompt = json_value(data, "prompt", std::string()); std::string prompt = json_value(data, "prompt", std::string());
std::vector<llama_tokens> tokenized_prompts = tokenize_input_prompts(ctx_server.ctx, prompt, true, true); std::vector<llama_tokens> tokenized_prompts = tokenize_input_prompts(ctx_server.ctx, prompt, true, true);
SRV_DBG("creating infill tasks, n_prompts = %d\n", (int) tokenized_prompts.size()); SRV_DBG("creating infill tasks, n_prompts = %d\n", (int) tokenized_prompts.size());
data["prompts"] = format_infill( data["prompt"] = format_infill(
ctx_server.ctx, ctx_server.ctx,
data.at("input_prefix"), data.at("input_prefix"),
data.at("input_suffix"), data.at("input_suffix"),