Update generation_test_hf_script.py

generation_test_hf_script.py

@@ -39,7 +39,7 @@ def run_test(model_name, test_ds):
     for i, entries in enumerate(test_ds):
 
         # prepare prompt packaging used in fine-tuning process
-        new_prompt = "<human>: " + entries["context"] + "\n" + entries["query"] + "\n" + "<bot>:"
+        new_prompt = "<human>: " + entries["context"] + "\n" + entries["query"] + "\n" + "<bot>:" + "\n"
 
         inputs = tokenizer(new_prompt, return_tensors="pt")
         start_of_output = len(inputs.input_ids[0])