From 545700ad409d66ba4d44415692c94a885d6a1bc6 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Wed, 26 Feb 2025 11:38:37 +0100 Subject: [PATCH] mod allegro --- allegro.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/allegro.py b/allegro.py index 6eef052..39e8018 100644 --- a/allegro.py +++ b/allegro.py @@ -111,8 +111,8 @@ def main(): print("Rozpoczęcie treningu...") trainer.train() - trainer.save_model("./trained_model/gpt") - tokenizer.save_pretrained("./trained_model/gpt") + trainer.save_model("./trained_model/allegro") + tokenizer.save_pretrained("./trained_model/allegro") if __name__ == "__main__": main() \ No newline at end of file