POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod allegro

This commit is contained in:
l.gabrysiak 2025-02-28 22:14:00 +01:00
parent ad00842f91
commit 4bcb4f2f5a
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
allegro.py
View File

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