diff --git a/allegro.py b/allegro.py index d5a1fe6..5f221f1 100644 --- a/allegro.py +++ b/allegro.py @@ -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!") \ No newline at end of file +print("✅ Model został wytrenowany i zapisany!") \ No newline at end of file