lphatnguyen/distillation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update easydistill/mmkd/infer.py

Browse Source 
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
bug-orz
2025-07-24 13:39:44 +08:00
committed by GitHub
parent fe2d2f5cdf
commit 56ba54f548
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
easydistill/mmkd/infer.py
View File

@@ -205,7 +205,7 @@ def generate_teacher_logits_batch(processor, llm, data_list, config, batch_size=
for k,v in pos.items(): for k,v in pos.items():
pos[k]=math.exp(v.logprob) 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 row in logits:
#for item in row: #for item in row:
writer.write(row) writer.write(row)