POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod gpt save tokenizer

This commit is contained in:
l.gabrysiak 2025-02-26 10:06:04 +01:00
parent f49fd406c5
commit 2a4404d708
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
gpt.py
View File

@ -100,6 +100,7 @@ def main():
print("Rozpoczęcie treningu...") print("Rozpoczęcie treningu...")
trainer.train() trainer.train()
trainer.save_model("./trained_model/gpt") trainer.save_model("./trained_model/gpt")
tokenizer.save_pretrained("./trained_model/gpt")
if __name__ == "__main__": if __name__ == "__main__":
main() main()