Merge pull request #7 from rhymes-ai/save_processor

support saving processor
rhymes-ai · Oct 3, 2024 · 8d4cb04 · 8d4cb04
2 parents a603434 + f7e7c3b
commit 8d4cb04
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)