Fix to use Path obj

This commit is contained in:
litagin02
2024-02-23 16:19:29 +09:00
parent 337ca71ec2
commit 0a6e4f1203

View File

@@ -1,10 +1,13 @@
"""
API server for TTS
TODO: server_editor.pyと統合する?
"""
import argparse
import os
import sys
from io import BytesIO
from pathlib import Path
from typing import Dict, Optional, Union
from urllib.parse import unquote
@@ -53,10 +56,8 @@ def load_models(model_holder: ModelHolder):
for model_name, model_paths in model_holder.model_files_dict.items():
model = Model(
model_path=model_paths[0],
config_path=os.path.join(model_holder.root_dir, model_name, "config.json"),
style_vec_path=os.path.join(
model_holder.root_dir, model_name, "style_vectors.npy"
),
config_path=model_holder.root_dir / model_name / "config.json",
style_vec_path=model_holder.root_dir / model_name / "style_vectors.npy",
device=model_holder.device,
)
model.load_net_g()
@@ -76,7 +77,7 @@ if __name__ == "__main__":
else:
device = "cuda" if torch.cuda.is_available() else "cpu"
model_dir = args.dir
model_dir = Path(args.dir)
model_holder = ModelHolder(model_dir, device)
if len(model_holder.model_names) == 0:
logger.error(f"Models not found in {model_dir}.")
@@ -105,9 +106,12 @@ if __name__ == "__main__":
request: Request,
text: str = Query(..., min_length=1, max_length=limit, description=f"セリフ"),
encoding: str = Query(None, description="textをURLデコードする(ex, `utf-8`)"),
model_id: int = Query(0, description="モデルID。`GET /models/info`のkeyの値を指定ください"),
model_id: int = Query(
0, description="モデルID。`GET /models/info`のkeyの値を指定ください"
),
speaker_name: str = Query(
None, description="話者名(speaker_idより優先)。esd.listの2列目の文字列を指定"
None,
description="話者名(speaker_idより優先)。esd.listの2列目の文字列を指定",
),
speaker_id: int = Query(
0, description="話者ID。model_assets>[model]>config.json内のspk2idを確認"
@@ -116,12 +120,17 @@ if __name__ == "__main__":
DEFAULT_SDP_RATIO,
description="SDP(Stochastic Duration Predictor)/DP混合比。比率が高くなるほどトーンのばらつきが大きくなる",
),
noise: float = Query(DEFAULT_NOISE, description="サンプルノイズの割合。大きくするほどランダム性が高まる"),
noise: float = Query(
DEFAULT_NOISE,
description="サンプルノイズの割合。大きくするほどランダム性が高まる",
),
noisew: float = Query(
DEFAULT_NOISEW, description="SDPイズ。大きくするほど発音の間隔にばらつきが出やすくなる"
DEFAULT_NOISEW,
description="SDPイズ。大きくするほど発音の間隔にばらつきが出やすくなる",
),
length: float = Query(
DEFAULT_LENGTH, description="話速。基準は1で大きくするほど音声は長くなり読み上げが遅まる"
DEFAULT_LENGTH,
description="話速。基準は1で大きくするほど音声は長くなり読み上げが遅まる",
),
language: Languages = Query(ln, description=f"textの言語"),
auto_split: bool = Query(DEFAULT_LINE_SPLIT, description="改行で分けて生成"),
@@ -129,20 +138,25 @@ if __name__ == "__main__":
DEFAULT_SPLIT_INTERVAL, description="分けた場合に挟む無音の長さ(秒)"
),
assist_text: Optional[str] = Query(
None, description="このテキストの読み上げと似た声音・感情になりやすくなる。ただし抑揚やテンポ等が犠牲になる傾向がある"
None,
description="このテキストの読み上げと似た声音・感情になりやすくなる。ただし抑揚やテンポ等が犠牲になる傾向がある",
),
assist_text_weight: float = Query(
DEFAULT_ASSIST_TEXT_WEIGHT, description="assist_textの強さ"
),
style: Optional[Union[int, str]] = Query(DEFAULT_STYLE, description="スタイル"),
style_weight: float = Query(DEFAULT_STYLE_WEIGHT, description="スタイルの強さ"),
reference_audio_path: Optional[str] = Query(None, description="スタイルを音声ファイルで行う"),
reference_audio_path: Optional[str] = Query(
None, description="スタイルを音声ファイルで行う"
),
):
"""Infer text to speech(テキストから感情付き音声を生成する)"""
logger.info(
f"{request.client.host}:{request.client.port}/voice { unquote(str(request.query_params) )}"
)
if model_id >= len(model_holder.models): # /models/refresh があるためQuery(le)で表現不可
if model_id >= len(
model_holder.models
): # /models/refresh があるためQuery(le)で表現不可
raise_validation_error(f"model_id={model_id} not found", "model_id")
model = model_holder.models[model_id]