mod allegro

This commit is contained in:
l.gabrysiak 2025-02-26 11:38:37 +01:00
parent b30937ceb4
commit 545700ad40
1 changed files with 2 additions and 2 deletions

View File

@ -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()