fac4f9a8ab
"text_processing" is clearer, but the import statement is longer. "nlp" is shorter and makes it clear that it is natural language processing.
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
from enum import StrEnum
|
|
from pathlib import Path
|
|
|
|
|
|
# Style-Bert-VITS2 のバージョン
|
|
LATEST_VERSION = "2.4"
|
|
|
|
# Style-Bert-VITS2 のベースディレクトリ
|
|
BASE_DIR = Path(__file__).parent.parent
|
|
|
|
# 利用可能な言語
|
|
## JP-Extra モデル利用時は JP 以外の言語の音声合成はできない
|
|
class Languages(StrEnum):
|
|
JP = "JP"
|
|
EN = "EN"
|
|
ZH = "ZH"
|
|
|
|
# 言語ごとのデフォルトの BERT トークナイザーのパス
|
|
DEFAULT_BERT_TOKENIZER_PATHS = {
|
|
Languages.JP: BASE_DIR / "bert" / "deberta-v2-large-japanese-char-wwm",
|
|
Languages.EN: BASE_DIR / "bert" / "deberta-v3-large",
|
|
Languages.ZH: BASE_DIR / "bert" / "chinese-roberta-wwm-ext-large",
|
|
}
|
|
|
|
# デフォルトのユーザー辞書ディレクトリ
|
|
## style_bert_vits2.nlp.japanese.user_dict モジュールのデフォルト値として利用される
|
|
## ライブラリとしての利用などで外部のユーザー辞書を指定したい場合は、user_dict 以下の各関数の実行時、引数に辞書データファイルのパスを指定する
|
|
DEFAULT_USER_DICT_DIR = BASE_DIR / "dict_data"
|
|
|
|
# デフォルトの推論パラメータ
|
|
DEFAULT_STYLE = "Neutral"
|
|
DEFAULT_STYLE_WEIGHT = 5.0
|
|
DEFAULT_SDP_RATIO = 0.2
|
|
DEFAULT_NOISE = 0.6
|
|
DEFAULT_NOISEW = 0.8
|
|
DEFAULT_LENGTH = 1.0
|
|
DEFAULT_LINE_SPLIT = True
|
|
DEFAULT_SPLIT_INTERVAL = 0.5
|
|
DEFAULT_ASSIST_TEXT_WEIGHT = 0.7
|
|
DEFAULT_ASSIST_TEXT_WEIGHT = 1.0
|
|
|
|
# Gradio のテーマ
|
|
## Built-in theme: "default", "base", "monochrome", "soft", "glass"
|
|
## See https://huggingface.co/spaces/gradio/theme-gallery for more themes
|
|
GRADIO_THEME = "NoCrypt/miku"
|