mod herbert

This commit is contained in:
l.gabrysiak 2025-03-01 00:26:32 +01:00
parent 4f486f021b
commit a5e5401548
1 changed files with 2 additions and 2 deletions

View File

@ -113,7 +113,7 @@ trainer = Trainer(
trainer.train() trainer.train()
# 1⃣1⃣ Zapis modelu # 1⃣1⃣ Zapis modelu
model.save_pretrained("./trained_model/gemma") model.save_pretrained("./models/herbert")
tokenizer.save_pretrained("./trained_model/gemma") tokenizer.save_pretrained("./models/herbert")
print("✅ Model został wytrenowany i zapisany!") print("✅ Model został wytrenowany i zapisany!")