File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -142,14 +142,14 @@ int main(int argc, char ** argv) {
142142 for (const auto & prompt : prompts) {
143143 std::vector<llama_token> inp;
144144
145- // split classification sequences and insert expected separator tokens
145+ // split classification pairs and insert expected separator tokens
146146 if (pooling_type == LLAMA_POOLING_TYPE_RANK && prompt.find (params.cls_sep ) != std::string::npos) {
147- std::vector<std::string> sequences = split_lines (prompt, params.cls_sep );
147+ std::vector<std::string> pairs = split_lines (prompt, params.cls_sep );
148148 std::string final_prompt;
149149
150- for (size_t i = 0 ; i < sequences .size (); i++) {
151- final_prompt += sequences [i];
152- if (i != sequences .size () - 1 ) {
150+ for (size_t i = 0 ; i < pairs .size (); i++) {
151+ final_prompt += pairs [i];
152+ if (i != pairs .size () - 1 ) {
153153 if (!added_eos_token.empty ()) {
154154 final_prompt += added_eos_token;
155155 }
You can’t perform that action at this time.
0 commit comments