From 56ba54f548680de3b22f491da446065a92a2c25f Mon Sep 17 00:00:00 2001 From: bug-orz Date: Thu, 24 Jul 2025 13:39:44 +0800 Subject: [PATCH] Update easydistill/mmkd/infer.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- easydistill/mmkd/infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/easydistill/mmkd/infer.py b/easydistill/mmkd/infer.py index 02add50..cd34ddd 100644 --- a/easydistill/mmkd/infer.py +++ b/easydistill/mmkd/infer.py @@ -205,7 +205,7 @@ def generate_teacher_logits_batch(processor, llm, data_list, config, batch_size= for k,v in pos.items(): pos[k]=math.exp(v.logprob) - with jsonlines.open(config["dataset"]["logits_path"], mode='a') as writer: + with jsonlines.open(config["dataset"]["logits_path"], mode='w') as writer: for row in logits: #for item in row: writer.write(row)