mirror of https://github.com/coqui-ai/TTS.git
19 lines
415 B
Python
19 lines
415 B
Python
import re
|
|
|
|
from TTS.speaker_encoder.model import SpeakerEncoder
|
|
|
|
|
|
def to_camel(text):
|
|
text = text.capitalize()
|
|
return re.sub(r"(?!^)_([a-zA-Z])", lambda m: m.group(1).upper(), text)
|
|
|
|
|
|
def setup_model(c):
|
|
model = SpeakerEncoder(
|
|
c.model_params["input_dim"],
|
|
c.model_params["proj_dim"],
|
|
c.model_params["lstm_dim"],
|
|
c.model_params["num_lstm_layers"],
|
|
)
|
|
return model
|