diff --git a/pred.py b/pred.py index ae1ec6fae31a86d5576f09493cd0c05f482403bd..990817fb509f630afd71c890d097303d661e9384 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