github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-06-23 14:07:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add new reverse prompt for new localdocs context feature.

Browse Source
This commit is contained in:
Adam Treat 2023-05-20 20:02:14 -04:00 committed by AT
parent 29b226405f
commit 80024a029c
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gpt4all-backend/gptj.cpp
View File

@ -983,7 +983,7 @@ void GPTJ::prompt(const std::string &prompt,
std::string cachedResponse; std::string cachedResponse;
std::vector<gpt_vocab::id> cachedTokens; std::vector<gpt_vocab::id> cachedTokens;
std::unordered_set<std::string> reversePrompts std::unordered_set<std::string> reversePrompts
= { "### Instruction", "### Prompt", "### Response", "### Human", "### Assistant" }; = { "### Instruction", "### Prompt", "### Response", "### Human", "### Assistant", "### Context" };
// predict next tokens // predict next tokens
int32_t totalPredictions = 0; int32_t totalPredictions = 0;

2
gpt4all-backend/llamamodel.cpp
View File

@ -179,7 +179,7 @@ void LLamaModel::prompt(const std::string &prompt,
std::string cachedResponse; std::string cachedResponse;
std::vector<llama_token> cachedTokens; std::vector<llama_token> cachedTokens;
std::unordered_set<std::string> reversePrompts std::unordered_set<std::string> reversePrompts
= { "### Instruction", "### Prompt", "### Response", "### Human", "### Assistant" }; = { "### Instruction", "### Prompt", "### Response", "### Human", "### Assistant", "### Context" };
// predict next tokens // predict next tokens
int32_t totalPredictions = 0; int32_t totalPredictions = 0;

2
gpt4all-backend/mpt.cpp
View File

@ -908,7 +908,7 @@ void MPT::prompt(const std::string &prompt,
std::string cachedResponse; std::string cachedResponse;
std::vector<int> cachedTokens; std::vector<int> cachedTokens;
std::unordered_set<std::string> reversePrompts std::unordered_set<std::string> reversePrompts
= { "### Instruction", "### Prompt", "### Response", "### Human", "### Assistant" }; = { "### Instruction", "### Prompt", "### Response", "### Human", "### Assistant", "### Context" };
// predict next tokens // predict next tokens
int32_t totalPredictions = 0; int32_t totalPredictions = 0;