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

Fix: In the “Merge” tab of the Web UI, re-create the TTSModelHolder using the passed TTSModelHolder instance variable so that ONNX models do not get mixed up

Browse Source
This commit is contained in:
tsukumi
2024-09-23 08:13:16 +09:00
parent b7cd512978
commit 245bfe54ae
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
gradio_tabs/merge.py
View File

@@ -1003,6 +1003,11 @@ def method_change(x: str):
def create_merge_app(model_holder: TTSModelHolder) -> gr.Blocks:
# ONNX モデルが混じらないよう、渡された TTSModelHolder のインスタンス変数を使って TTSModelHolder を作り直す
model_holder = TTSModelHolder(
model_holder.root_dir, model_holder.device, model_holder.onnx_providers, ignore_onnx=True
)
model_names = model_holder.model_names
if len(model_names) == 0:
logger.error(
@@ -1527,7 +1532,7 @@ def create_merge_app(model_holder: TTSModelHolder) -> gr.Blocks:
if __name__ == "__main__":
device = "cuda" if torch.cuda.is_available() else "cpu"
model_holder = TTSModelHolder(
assets_root, device, torch_device_to_onnx_providers(device)
assets_root, device, torch_device_to_onnx_providers(device), ignore_onnx=True
)
app = create_merge_app(model_holder)
app.launch(inbrowser=True)