mod allegro
This commit is contained in:
parent
b30937ceb4
commit
545700ad40
|
|
@ -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()
|
||||
Loading…
Reference in New Issue