bug fixed for get_pred

bigai-nlco · Feb 1, 2024 · c231289 · c231289
1 parent 351153c
commit c231289
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Prediction/pred_gpt_models.py b/Prediction/pred_gpt_models.py
@@ -71,7 +71,7 @@ def get_pred(model, data_instance, tokenizer, max_length, max_gen, prompt_format
         pred = rsp['choices'][0]['message']['content']
 
         ans.append(pred)
-        groundtruth.append(raw_inputs)
+        groundtruth.append(data_instance["output"])
 
     else:
         preds['qa_pairs'] = eval(data_instance['qa_pairs'])

diff --git a/Prediction/pred_opensource_models.py b/Prediction/pred_opensource_models.py
@@ -60,7 +60,7 @@ def get_pred(model, data_instance, tokenizer, max_length, max_gen, prompt_format
         pred = tokenizer.decode(output[context_length:], skip_special_tokens=True)
 
         ans.append(pred)
-        groundtruth.append(raw_inputs)
+        groundtruth.append(data_instance["output"])
 
     else:
         preds['qa_pairs'] = eval(data_instance['qa_pairs'])