ably.do/allegro.py

10 lines
363 B
Python
Raw Normal View History

2025-02-28 16:12:04 -05:00
from transformers import MarianForCausalLM, MarianTokenizer, TrainingArguments
2025-02-28 15:26:21 -05:00
model_name = "allegro/multislav-5lang"
2025-02-28 15:41:23 -05:00
model = MarianForCausalLM.from_pretrained(model_name)
tokenizer = MarianTokenizer.from_pretrained(model_name)
2025-02-28 15:26:21 -05:00
2025-02-28 16:09:57 -05:00
model.save_pretrained("./models/ably")
tokenizer.save_pretrained("./models/ably")
2025-02-28 16:12:04 -05:00
print("✅ Model został wytrenowany i zapisany!")