diff --git a/recipes/distilqwen_series/distillqwen2/dpo_student_infer_only.py b/recipes/distilqwen_series/distillqwen2/dpo_student_infer_only.py index a7cb65e..d006c1c 100644 --- a/recipes/distilqwen_series/distillqwen2/dpo_student_infer_only.py +++ b/recipes/distilqwen_series/distillqwen2/dpo_student_infer_only.py @@ -32,9 +32,7 @@ def read_json_field(filename): data = json.load(file) output = [] for item in data: - instruction = item["instruction"] - output = item["output"] - output.append({"prompt": instruction, "chosen": output}) + output.append({"prompt": item["instruction"], "chosen": item["output"]}) return output except FileNotFoundError: logging.error("The file was not found.")