This commit is contained in:
Andrii Pervashov 2024-11-08 17:47:35 +01:00
parent e5673fde0a
commit 24312af4d9
2 changed files with 1 additions and 1 deletions

View File

@ -4,7 +4,7 @@ import jiwer
import sacrebleu import sacrebleu
# 1. Загрузка модели и токенизатора # 1. Загрузка модели и токенизатора
model_path = "T5Autocorrection1337" # Укажите путь к вашей модели model_path = "google/mt5-large" # Укажите путь к вашей модели
tokenizer = AutoTokenizer.from_pretrained(model_path) tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSeq2SeqLM.from_pretrained(model_path) model = AutoModelForSeq2SeqLM.from_pretrained(model_path)