tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor: introducing Ruff

Browse Source
This commit is contained in:
tsukumi
2024-03-07 19:52:34 +00:00
parent 3f07c256e3
commit 4d5c537f95
8 changed files with 26 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
losses.py
View File

@@ -2,8 +2,6 @@ import torch
import torchaudio
from transformers import AutoModel
from style_bert_vits2.logging import logger
def feature_loss(fmap_r, fmap_g):
loss = 0

9
pyproject.toml Normal file
View File

@@ -0,0 +1,9 @@
[tool.ruff]
# インデント幅を 4 に設定
indent-width = 4
# 行の長さを 100 文字に設定
line-length = 100
# Python 3.10 向けにフォーマット
target-version = "py310"

4
server_fastapi.py
View File

@@ -104,7 +104,7 @@ if __name__ == "__main__":
@app.get("/voice", response_class=AudioResponse)
async def voice(
request: Request,
text: str = Query(..., min_length=1, max_length=limit, description=f"セリフ"),
text: str = Query(..., min_length=1, max_length=limit, description="セリフ"),
encoding: str = Query(None, description="textをURLデコードする(ex, `utf-8`)"),
model_id: int = Query(
0, description="モデルID。`GET /models/info`のkeyの値を指定ください"
@@ -132,7 +132,7 @@ if __name__ == "__main__":
DEFAULT_LENGTH,
description="話速。基準は1で大きくするほど音声は長くなり読み上げが遅まる",
),
language: Languages = Query(ln, description=f"textの言語"),
language: Languages = Query(ln, description="textの言語"),
auto_split: bool = Query(DEFAULT_LINE_SPLIT, description="改行で分けて生成"),
split_interval: float = Query(
DEFAULT_SPLIT_INTERVAL, description="分けた場合に挟む無音の長さ（秒）"

1
style_bert_vits2/text_processing/english/__init__.py
View File

@@ -234,7 +234,6 @@ def refine_syllables(syllables):
return phonemes, tones
import re
import inflect
_inflect = inflect.engine()

4
style_bert_vits2/text_processing/japanese/__init__.py
View File

@@ -1,2 +1,2 @@
from style_bert_vits2.text_processing.japanese.g2p import g2p # type: ignore
from style_bert_vits2.text_processing.japanese.normalizer import normalize_text # type: ignore
from style_bert_vits2.text_processing.japanese.g2p import g2p # noqa: F401
from style_bert_vits2.text_processing.japanese.normalizer import normalize_text # noqa: F401

1
train_ms_jp_extra.py
View File

@@ -1,6 +1,5 @@
import argparse
import datetime
import gc
import os
import platform

6
webui_style_vectors.py
View File

@@ -153,7 +153,7 @@ def do_dbscan_gradio(eps=2.5, min_samples=15):
return [
plt,
gr.Slider(maximum=MAX_CLUSTER_NUM),
f"クラスタが数が0です。パラメータを変えてみてください。",
"クラスタが数が0です。パラメータを変えてみてください。",
] + [gr.Audio(visible=False)] * MAX_AUDIO_NUM
return [plt, gr.Slider(maximum=n_clusters, value=1), n_clusters] + [
@@ -212,7 +212,7 @@ def save_style_vectors_from_clustering(model_name, style_names_str: str):
if len(style_name_list) != len(centroids) + 1:
return f"スタイルの数が合いません。`,`で正しく{len(centroids)}個に区切られているか確認してください: {style_names_str}"
if len(set(style_names)) != len(style_names):
return f"スタイル名が重複しています。"
return "スタイル名が重複しています。"
logger.info(f"Backup {config_path} to {config_path}.bak")
shutil.copy(config_path, f"{config_path}.bak")
@@ -243,7 +243,7 @@ def save_style_vectors_from_files(
return f"音声ファイルとスタイル名の数が合いません。`,`で正しく{len(style_names)}個に区切られているか確認してください: {audio_files_str}と{style_names_str}"
style_name_list = [DEFAULT_STYLE] + style_names
if len(set(style_names)) != len(style_names):
return f"スタイル名が重複しています。"
return "スタイル名が重複しています。"
style_vectors = [mean]
wavs_dir = os.path.join(dataset_root, model_name, "wavs")

20
webui_train.py
View File

@@ -147,10 +147,10 @@ def resample(model_name, normalize, trim, num_processes):
cmd.append("--trim")
success, message = run_script_with_log(cmd)
if not success:
logger.error(f"Step 2: resampling failed.")
logger.error("Step 2: resampling failed.")
return False, f"Step 2, Error: 音声ファイルの前処理に失敗しました:\n{message}"
elif message:
logger.warning(f"Step 2: resampling finished with stderr.")
logger.warning("Step 2: resampling finished with stderr.")
return True, f"Step 2, Success: 音声ファイルの前処理が完了しました:\n{message}"
logger.success("Step 2: resampling finished.")
return True, "Step 2, Success: 音声ファイルの前処理が完了しました"
@@ -197,13 +197,13 @@ def preprocess_text(model_name, use_jp_extra, val_per_lang, yomi_error):
cmd.append("--use_jp_extra")
success, message = run_script_with_log(cmd)
if not success:
logger.error(f"Step 3: preprocessing text failed.")
logger.error("Step 3: preprocessing text failed.")
return (
False,
f"Step 3, Error: 書き起こしファイルの前処理に失敗しました:\n{message}",
)
elif message:
logger.warning(f"Step 3: preprocessing text finished with stderr.")
logger.warning("Step 3: preprocessing text finished with stderr.")
return (
True,
f"Step 3, Success: 書き起こしファイルの前処理が完了しました:\n{message}",
@@ -225,10 +225,10 @@ def bert_gen(model_name):
]
)
if not success:
logger.error(f"Step 4: bert_gen failed.")
logger.error("Step 4: bert_gen failed.")
return False, f"Step 4, Error: BERT特徴ファイルの生成に失敗しました:\n{message}"
elif message:
logger.warning(f"Step 4: bert_gen finished with stderr.")
logger.warning("Step 4: bert_gen finished with stderr.")
return (
True,
f"Step 4, Success: BERT特徴ファイルの生成が完了しました:\n{message}",
@@ -250,13 +250,13 @@ def style_gen(model_name, num_processes):
]
)
if not success:
logger.error(f"Step 5: style_gen failed.")
logger.error("Step 5: style_gen failed.")
return (
False,
f"Step 5, Error: スタイル特徴ファイルの生成に失敗しました:\n{message}",
)
elif message:
logger.warning(f"Step 5: style_gen finished with stderr.")
logger.warning("Step 5: style_gen finished with stderr.")
return (
True,
f"Step 5, Success: スタイル特徴ファイルの生成が完了しました:\n{message}",
@@ -350,10 +350,10 @@ def train(model_name, skip_style=False, use_jp_extra=True, speedup=False):
cmd.append("--speedup")
success, message = run_script_with_log(cmd, ignore_warning=True)
if not success:
logger.error(f"Train failed.")
logger.error("Train failed.")
return False, f"Error: 学習に失敗しました:\n{message}"
elif message:
logger.warning(f"Train finished with stderr.")
logger.warning("Train finished with stderr.")
return True, f"Success: 学習が完了しました:\n{message}"
logger.success("Train finished.")
return True, "Success: 学習が完了しました"