1
0
Fork 0
mirror of https://gitlab.com/niansa/discord_llama.git synced 2025-03-06 20:48:25 +01:00

Disallow {user_prompt} and {bot_prompt} formatters in non-instruct prompt files

This commit is contained in:
niansa/tuxifan 2023-05-15 08:03:09 +02:00
parent 8ec7908f10
commit 3011dadac3

View file

@ -205,7 +205,7 @@ private:
using namespace fmt::literals;
if (prompt.back() != '\n') prompt.push_back('\n');
llm->set_scroll_callback(scroll_cb);
co_await llm->append(fmt::format(fmt::runtime(prompt), "bot_name"_a=bot.me.username, "bot_prompt"_a=model_config.bot_prompt, "user_prompt"_a=model_config.user_prompt), show_console_progress);
co_await llm->append(fmt::format(fmt::runtime(prompt), "bot_name"_a=bot.me.username), show_console_progress);
// Serialize end result
std::ofstream f(filename, std::ios::binary);
co_await llm->serialize(f);