RUAccent-encoder

Эта модель представляет собой энкодер для TTS систем, который умеет разрешать омографию. Предназначен для интеграции в TTS системы, но лучше обратите внимание на энкодер который вогнали в стресс (как и автора RUAccent)

Использование

import torch
from transformers import AutoModel
import chartk
tokenizer = chartk.CharacterTokenizer.from_pretrained('ruaccent/RUAccent-encoder')
model = AutoModel.from_pretrained('ruaccent/RUAccent-stressed-encoder')
text = "На горе стоит замок"
inputs = tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=512)
with torch.no_grad():
    outputs = model(**inputs)
last_hidden_state = outputs.last_hidden_state