From 3a1dc7c58ffb42baac2f5d6807b1ee190efb9d47 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Mon, 26 Feb 2024 21:27:02 +0900 Subject: [PATCH] Fix using path instead of str --- webui_merge.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/webui_merge.py b/webui_merge.py index 7d11862..69f7e8d 100644 --- a/webui_merge.py +++ b/webui_merge.py @@ -2,6 +2,7 @@ import argparse import json import os import sys +from pathlib import Path import gradio as gr import numpy as np @@ -27,7 +28,7 @@ with open(os.path.join("configs", "paths.yml"), "r", encoding="utf-8") as f: # dataset_root = path_config["dataset_root"] assets_root = path_config["assets_root"] -model_holder = ModelHolder(assets_root, device) +model_holder = ModelHolder(Path(assets_root), device) def merge_style(model_name_a, model_name_b, weight, output_name, style_triple_list): @@ -226,7 +227,7 @@ def simple_tts(model_name, text, style=DEFAULT_STYLE, style_weight=1.0): config_path = os.path.join(assets_root, model_name, "config.json") style_vec_path = os.path.join(assets_root, model_name, "style_vectors.npy") - model = Model(model_path, config_path, style_vec_path, device) + model = Model(Path(model_path), Path(config_path), Path(style_vec_path), device) return model.infer(text, style=style, style_weight=style_weight)