Format import using isort --profile black --gitignore --lai 2 .

This commit is contained in:
litagin02
2024-03-11 12:37:51 +09:00
parent 164b5c4a85
commit dada0af2b6
29 changed files with 50 additions and 35 deletions

View File

@@ -1,4 +1,4 @@
from typing import Optional, TYPE_CHECKING
from typing import TYPE_CHECKING, Optional
from style_bert_vits2.constants import Languages
from style_bert_vits2.nlp.symbols import (
@@ -7,6 +7,7 @@ from style_bert_vits2.nlp.symbols import (
SYMBOLS,
)
# __init__.py は配下のモジュールをインポートした時点で実行される
# PyTorch のインポートは重いので、型チェック時以外はインポートしない
if TYPE_CHECKING:
@@ -99,10 +100,10 @@ def cleaned_text_to_sequence(
cleaned_phones: list[str], tones: list[int], language: Languages
) -> tuple[list[int], list[int], list[int]]:
"""
テキスト文字列を、テキスト内の記号に対応する一連の ID に変換する
音素リスト・アクセントリスト・言語を、テキスト内の対応する ID に変換する
Args:
cleaned_phones (list[str]): clean_text() でクリーニングされた音素のリスト (?)
cleaned_phones (list[str]): clean_text() でクリーニングされた音素のリスト
tones (list[int]): 各音素のアクセント
language (Languages): テキストの言語

View File

@@ -9,7 +9,7 @@ Style-Bert-VITS2 の学習・推論に必要な各言語ごとの BERT モデル
"""
import gc
from typing import cast, Optional, Union
from typing import Optional, Union, cast
import torch
from transformers import (

View File

@@ -2,7 +2,7 @@ import re
from pathlib import Path
import jieba.posseg as psg
from pypinyin import lazy_pinyin, Style
from pypinyin import Style, lazy_pinyin
from style_bert_vits2.nlp.chinese.tone_sandhi import ToneSandhi
from style_bert_vits2.nlp.symbols import PUNCTUATIONS

View File

@@ -13,8 +13,7 @@
# limitations under the License.
import jieba
from pypinyin import lazy_pinyin
from pypinyin import Style
from pypinyin import Style, lazy_pinyin
class ToneSandhi:

View File

@@ -1,5 +1,6 @@
import re
import unicodedata
from num2words import num2words
from style_bert_vits2.nlp.symbols import PUNCTUATIONS

View File

@@ -17,12 +17,13 @@ from fastapi import HTTPException
from style_bert_vits2.constants import DEFAULT_USER_DICT_DIR
from style_bert_vits2.nlp.japanese import pyopenjtalk_worker as pyopenjtalk
from style_bert_vits2.nlp.japanese.user_dict.word_model import UserDictWord, WordTypes
from style_bert_vits2.nlp.japanese.user_dict.part_of_speech_data import (
MAX_PRIORITY,
MIN_PRIORITY,
part_of_speech_data,
)
from style_bert_vits2.nlp.japanese.user_dict.word_model import UserDictWord, WordTypes
# root_dir = engine_root()
# save_dir = get_save_dir()

View File

@@ -14,6 +14,7 @@ from style_bert_vits2.nlp.japanese.user_dict.word_model import (
WordTypes,
)
MIN_PRIORITY = USER_DICT_MIN_PRIORITY
MAX_PRIORITY = USER_DICT_MAX_PRIORITY

View File

@@ -11,6 +11,7 @@ from typing import List, Optional
from pydantic import BaseModel, Field, validator
USER_DICT_MIN_PRIORITY = 0
USER_DICT_MAX_PRIORITY = 10