From f7e7c3b3a441424dba33bef239084b65afcd7391 Mon Sep 17 00:00:00 2001 From: aria-hacker Date: Thu, 3 Oct 2024 08:16:56 +0800 Subject: [PATCH] support saving processor --- aria/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aria/train.py b/aria/train.py index 6dad4bf..f6e1c3d 100644 --- a/aria/train.py +++ b/aria/train.py @@ -224,6 +224,7 @@ def main(): ) trainer.train(resume_from_checkpoint=training_args.resume_from_checkpoint) + processor.save_pretrained(training_args.output_dir) trainer.save_model(training_args.output_dir)