From bb22cd5048c48323eafb2fd19a9689bc81369850 Mon Sep 17 00:00:00 2001
From: bys0318 <bys0318@126.com>
Date: Wed, 2 Aug 2023 16:49:54 +0800
Subject: [PATCH] fix Chinese output
---
pred.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pred.py b/pred.py
index ae1ec6f..990817f 100644
--- a/pred.py
+++ b/pred.py
@@ -57,5 +57,5 @@ if __name__ == '__main__':
preds = get_pred(model, tokenizer, data, max_length, max_gen, prompt_format, dataset, device)
with open(f"pred/{dataset}.jsonl", "w") as f:
for pred in preds:
- json.dump(pred, f)
+ json.dump(pred, f, ensure_ascii=False)
f.write('\n')
\ No newline at end of file
--
GitLab