diff --git a/gpt.py b/gpt.py index f35a456..04efbdf 100644 --- 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() \ No newline at end of file