Refactor: moved the module for extracting BERT features from text in each language to style_bert_vits2/text_processing/(language)/bert_feature.py

This commit is contained in:
tsukumi
2024-03-07 03:32:07 +00:00
parent c3c0dd8b32
commit 62919e904e
16 changed files with 172 additions and 101 deletions

View File

@@ -11,9 +11,10 @@ import yaml
from safetensors import safe_open
from safetensors.torch import save_file
from common.tts_model import Model, ModelHolder
from style_bert_vits2.constants import DEFAULT_STYLE, GRADIO_THEME
from style_bert_vits2.logging import logger
from common.tts_model import Model, ModelHolder
voice_keys = ["dec"]
voice_pitch_keys = ["flow"]