Это доработанная версия Multilingual Bart, обученная русскому языку для исправления грамматических ошибок. из трансформеров import MBartForConditionalGeneration, MBart50TokenizerFast model = MBartForConditionalGeneration.from_pretrained(«MRNH/mbart-russian-grammar-corrector») tokenizer = MBart50TokenizerFast.frompretrained(«MRNH/mbart-russian-grammar-corrector», srclang=»ruRU», tgtlang=»ruRU») input = tokenizer(«Я был здесь вчера, чтобы учиться», texttarget=»Я был здесь вчера, чтобы учиться», return_tensors=’pt’) output = model.generate(input[«inputids»],attentionmask=input[«attentionmask»],forcedbostokenid=tokenizerit.langcodetoid[«ru_RU»]) Обучение модели выполняется с использованием следующего вычисления потерь на основе вывода скрытого состояния h: h.logits, h.loss = model(inputids=input[«inputids»], focusmask=input[«attentionmask»], labels=input[«labels»])
Модальности:
Генерация текста
Задача: Генерация текста
Автор: MRNH
Теги: mbart, text2text-generation, grammatical error correction, GEC, russian, ru, endpoints_compatible
Лайков: 6 | Загрузок: 221
Описание основано на материалах HuggingFace. Перевод выполнен автоматически.
