diff --git a/convert_onnx.py b/convert_onnx.py index b323728..1b1fe5d 100644 --- a/convert_onnx.py +++ b/convert_onnx.py @@ -45,7 +45,9 @@ if __name__ == "__main__": model_paths: list[Path] = [] if Path(args.model).is_dir(): for path in Path(args.model).glob("**/*.safetensors"): - model_paths.append(path) + # . から始まるファイルは除外 + if not path.name.startswith("."): + model_paths.append(path) else: model_paths.append(Path(args.model)) diff --git a/gradio_tabs/merge.py b/gradio_tabs/merge.py index 6082d1f..d55b04f 100644 --- a/gradio_tabs/merge.py +++ b/gradio_tabs/merge.py @@ -1005,7 +1005,10 @@ 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_holder.root_dir, + model_holder.device, + model_holder.onnx_providers, + ignore_onnx=True, ) model_names = model_holder.model_names diff --git a/style_bert_vits2/tts_model.py b/style_bert_vits2/tts_model.py index 47f9cc9..2f9e592 100644 --- a/style_bert_vits2/tts_model.py +++ b/style_bert_vits2/tts_model.py @@ -523,7 +523,8 @@ class TTSModelHolder: [ f for f in model_dir.iterdir() - if f.suffix in suffixes + # 上記 suffixes にマッチするファイルのみを取得し、. から始まるファイルは除外 + if f.suffix in suffixes and not f.name.startswith(".") ] ) if len(model_files) == 0: