This commit is contained in:
litagin02
2024-02-25 22:28:02 +09:00
parent d3a5cb8569
commit f749950be2
4 changed files with 45 additions and 5 deletions

View File

@@ -16,6 +16,7 @@ import zipfile
from datetime import datetime
from io import BytesIO
from pathlib import Path
import yaml
import numpy as np
import pyopenjtalk
@@ -151,6 +152,12 @@ origins = [
"http://127.0.0.1:8000",
]
# Get path settings
with open(Path("configs/paths.yml"), "r", encoding="utf-8") as f:
path_config: dict[str, str] = yaml.safe_load(f.read())
# dataset_root = path_config["dataset_root"]
assets_root = path_config["assets_root"]
parser = argparse.ArgumentParser()
parser.add_argument("--model_dir", type=str, default="model_assets/")
parser.add_argument("--device", type=str, default="cuda")
@@ -158,6 +165,9 @@ parser.add_argument("--port", type=int, default=8000)
parser.add_argument("--inbrowser", action="store_true")
parser.add_argument("--line_length", type=int, default=None)
parser.add_argument("--line_count", type=int, default=None)
parser.add_argument(
"--dir", "-d", type=str, help="Model directory", default=assets_root
)
args = parser.parse_args()
device = args.device
@@ -296,7 +306,11 @@ def multi_synthesis(request: MultiSynthesisRequest):
)
audios = []
for i, req in enumerate(lines):
# Loade model
if args.line_length is not None and len(req.text) > args.line_length:
raise HTTPException(
status_code=400,
detail=f"1行の文字数は{args.line_length}文字以下にしてください。",
)
try:
model = model_holder.load_model(
model_name=req.model, model_path_str=req.modelFile