From 4bcb4f2f5aa9e42019c07f98e590154f8cdb5bd0 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Fri, 28 Feb 2025 22:14:00 +0100 Subject: [PATCH] mod allegro --- allegro.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/allegro.py b/allegro.py index 53b9036..5f221f1 100644 --- a/allegro.py +++ b/allegro.py @@ -1,8 +1,7 @@ -from transformers import MarianForCausalLM, MarianTokenizer, TrainingArguments +from transformers import AutoTokenizer, AutoModelForSeq2SeqLM -model_name = "allegro/multislav-5lang" -model = MarianForCausalLM.from_pretrained(model_name) -tokenizer = MarianTokenizer.from_pretrained(model_name) +model = AutoModelForSeq2SeqLM.from_pretrained("allegro/multislav-5lang") +tokenizer = AutoTokenizer.from_pretrained("allegro/multislav-5lang") model.save_pretrained("./models/ably") tokenizer.save_pretrained("./models/ably")