mod allegro
This commit is contained in:
parent
b30937ceb4
commit
545700ad40
|
|
@ -111,8 +111,8 @@ 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/allegro")
|
||||||
tokenizer.save_pretrained("./trained_model/gpt")
|
tokenizer.save_pretrained("./trained_model/allegro")
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
Loading…
Reference in New Issue