tiny fix

Former-commit-id: 994f2a79831a1dea8425e3eff62f0bc8238b78d6
2023-06-27 23:54:24 +08:00
parent 2d22961c7d
commit e19dcc13e3
4 changed files with 15 additions and 5 deletions
--- a/src/export_model.py
+++ b/src/export_model.py
@@ -13,6 +13,7 @@ def main():
    model.save_pretrained(training_args.output_dir, max_shard_size="10GB")
    tokenizer.save_pretrained(training_args.output_dir)
    print("model and tokenizer have been saved at:", training_args.output_dir)
+    print("Remember to copy the *.py files from the original directory.")


 if __name__ == "__main__":