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!")
|