From 345049abfc0c64eebd7f51ec02c0957badf17716 Mon Sep 17 00:00:00 2001 From: Vidalnt Date: Wed, 4 Dec 2024 18:25:19 -0500 Subject: [PATCH] fix: :bug: Minor errors --- rvc/lib/algorithm/synthesizers.py | 1 + tabs/inference/inference.py | 20 +++++++++++--------- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/rvc/lib/algorithm/synthesizers.py b/rvc/lib/algorithm/synthesizers.py index e69f08eb..e0fd1db2 100644 --- a/rvc/lib/algorithm/synthesizers.py +++ b/rvc/lib/algorithm/synthesizers.py @@ -46,6 +46,7 @@ def __init__( n_layers: int, kernel_size: int, p_dropout: float, + resblock: str, resblock_kernel_sizes: list, resblock_dilation_sizes: list, upsample_rates: list, diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py index f6e03800..e11376d9 100644 --- a/tabs/inference/inference.py +++ b/tabs/inference/inference.py @@ -178,7 +178,7 @@ def change_choices(model): if model: speakers = get_speakers_id(model) else: - speakers = 0 + speakers = [0] names = [ os.path.join(root, file) for root, _, files in os.walk(model_root_relative, topdown=False) @@ -213,7 +213,7 @@ def change_choices(model): "choices": ( sorted(speakers) if speakers is not None and isinstance(speakers, (list, tuple)) - else [] + else [0] ), "__type__": "update", }, @@ -221,7 +221,7 @@ def change_choices(model): "choices": ( sorted(speakers) if speakers is not None and isinstance(speakers, (list, tuple)) - else [] + else [0] ), "__type__": "update", }, @@ -322,16 +322,18 @@ def refresh_embedders_folders(): def get_speakers_id(model): if model: - model_data = torch.load(os.path.join(now_dir, model), map_location="cpu") - speakers_id = model_data.get("speakers_id") - if speakers_id: - return list(range(speakers_id)) - else: + try: + model_data = torch.load(os.path.join(now_dir, model), map_location="cpu") + speakers_id = model_data.get("speakers_id") + if speakers_id: + return list(range(speakers_id)) + else: + return [0] + except Exception as e: return [0] else: return [0] - # Inference tab def inference_tab(): default_weight = names[0] if names else None