support falcon model #72

Former-commit-id: 72cc3ff0e6de641073de1159196319705f8efe85
2023-07-05 15:00:06 +08:00
parent 4b093996a7
commit f1de82f08e
5 changed files with 46 additions and 29 deletions
--- a/src/export_model.py
+++ b/src/export_model.py
@@ -13,7 +13,6 @@ 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__":