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

Update cvector-generator.cpp

Browse source
This commit is contained in:
HatsuneMikuUwU33 2024-06-21 11:10:16 +02:00 • committed by GitHub
parent fe8205ba1c
commit 6896afb23a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/cvector-generator/cvector-generator.cpp
View file

@ -378,7 +378,7 @@ static int prepare_entries(gpt_params & params, train_context & ctx_train) {
std::vector<std::string> completions = ctrlvec_load_prompt_file(params.cvector_completions_file, false);
auto format_template = [](std::string persona, std::string suffix) {
// entry in positive/negative.txt must already be formatted i.e. "[INST] Act as if you're extremely happy. [/INST]"
return persona + " " + suffix;
return persona + suffix;
};
for (size_t i = 0; i < positive_prompts.size(); ++i) {
for (int j = 0; j < std::min((int) completions.size(), params.n_completions); ++j) {