Merge pull request #313 from maybeluo/main

writing generated result with utf-8

ptuning/main.py

@@ -374,9 +374,10 @@ def main():
                 )
                 labels = [label.strip() for label in labels]
                 output_prediction_file = os.path.join(training_args.output_dir, "generated_predictions.txt")
-                with open(output_prediction_file, "w") as writer:
+                with open(output_prediction_file, "w", encoding="utf-8") as writer:
                     for p, l in zip(predictions, labels):
-                        writer.write(json.dumps({"labels": l, "predict": p}, ensure_ascii=False))
+                        res = json.dumps({"labels": l, "predict": p}, ensure_ascii=False)
+                        writer.write(f"{res}\n")
     return results
 
 
@@ -386,4 +387,4 @@ def _mp_fn(index):
 
 
 if __name__ == "__main__":
-    main()
+    main()