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 23:04:03 +01:00
parent e3a94fa5ae
commit d9541a9a28
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
allegro.py
View File

@ -1,11 +1,9 @@
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
# Załaduj model z formatu safetensors
model = AutoModelForSeq2SeqLM.from_pretrained("ścieżka_do_modelu", from_tf=False, from_safetensors=True)
tokenizer = AutoTokenizer.from_pretrained("ścieżka_do_modelu")
model = AutoModelForSeq2SeqLM.from_pretrained("allegro/multislav-5lang")
tokenizer = AutoTokenizer.from_pretrained("allegro/multislav-5lang")
# Zapisz model i tokenizer w formacie kompatybilnym z PyTorch (plik .bin)
model.save_pretrained("./models/ably")
tokenizer.save_pretrained("./models/ably")
print("✅ Model zapisany w formacie binarnym!")
print("✅ Model został wytrenowany i zapisany!")