RUAccent-encoders
Collection
An encoder models for TTS tasks
•
2 items
•
Updated
•
2
Эта модель представляет собой дистиллированную версию StressCDLM, специально обученную char level модель для экспериментов, когда модели нужно видеть символы и отфайнтюненую на корпусе ударений.
RUAccent-stressed-encoder - это энкодер, оптимизированный для обработки русского текста с ударениями. Предназначен для интеграции в качестве текстового энкодера в TTS системы.
import torch
from transformers import AutoModel
import chartk
tokenizer = chartk.CharacterTokenizer.from_pretrained('ruaccent/RUAccent-stressed-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