|
import importlib |
|
import re |
|
from typing import Dict, List, Union |
|
|
|
|
|
def to_camel(text): |
|
text = text.capitalize() |
|
return re.sub(r"(?!^)_([a-zA-Z])", lambda m: m.group(1).upper(), text) |
|
|
|
|
|
def setup_model(config: "Coqpit", samples: Union[List[List], List[Dict]] = None) -> "BaseVC": |
|
print(" > Using model: {}".format(config.model)) |
|
|
|
if "model" in config and config["model"].lower() == "freevc": |
|
MyModel = importlib.import_module("TTS.vc.models.freevc").FreeVC |
|
model = MyModel.init_from_config(config, samples) |
|
return model |
|
|
