cvector-generator: Moe Moe Fixie-Fixie for Lots of Formats~! ♡(ᐢ ᴥ ᐢ)♡ (#8052)

* Update negative.txt

* Update positive.txt

* Update cvector-generator.cpp

* Update cvector-generator.cpp

diff --git a/examples/cvector-generator/cvector-generator.cpp b/examples/cvector-generator/cvector-generator.cpp
index 9941683db677e65205ca778f9741c73e844c6dca..8653bb63052812bff0f42e444769889bf263918f 100644 (file)
--- a/examples/cvector-generator/cvector-generator.cpp
+++ b/examples/cvector-generator/cvector-generator.cpp
@@ -377,8 +377,8 @@ static int prepare_entries(gpt_params & params, train_context & ctx_train) {
     // create templated prompts
     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;
+        // entry in positive/negative.txt must already be formatted i.e. "[INST] Act as if you're extremely happy. [/INST] "
+        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) {

diff --git a/examples/cvector-generator/negative.txt b/examples/cvector-generator/negative.txt
index 2ac3387f184b06f647c632fe061bb8fb8ea6b705..3e9951752e8862f4002a3e535a0199d32a06a983 100644 (file)
--- a/examples/cvector-generator/negative.txt
+++ b/examples/cvector-generator/negative.txt
@@ -1 +1 @@
-[INST] Act like a person who is extremely sad. [/INST]
\ No newline at end of file
+[INST] Act like a person who is extremely sad. [/INST] 

diff --git a/examples/cvector-generator/positive.txt b/examples/cvector-generator/positive.txt
index f28e9aa1aeb72a35d476ee450b872b4a78d8943e..8802367873cd993e10a8a96f7d8b9b2c5be59203 100644 (file)
--- a/examples/cvector-generator/positive.txt
+++ b/examples/cvector-generator/positive.txt
@@ -1 +1 @@
-[INST] Act like a person who is extremely happy. [/INST]
\ No newline at end of file
+[INST] Act like a person who is extremely happy. [/INST]