mod gpt save tokenizer

2025-02-26 10:06:04 +01:00 · 2025-02-26 10:06:04 +01:00 · 2a4404d708
parent f49fd406c5
commit 2a4404d708
1 changed files with 1 additions and 0 deletions
--- a/gpt.py
+++ b/gpt.py
@ -100,6 +100,7 @@ def main():
    print("Rozpoczęcie treningu...")
    trainer.train()
    trainer.save_model("./trained_model/gpt")
+    tokenizer.save_pretrained("./trained_model/gpt")

 if __name__ == "__main__":
    main()