From 02d739913c96c8e792c953a79cac210a04232d8b Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Wed, 4 Dec 2024 21:43:37 +0000 Subject: [PATCH] chore(format): run black on main --- tabs/inference/inference.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py index bced5ffc..f6e03800 100644 --- a/tabs/inference/inference.py +++ b/tabs/inference/inference.py @@ -322,7 +322,7 @@ def refresh_embedders_folders(): def get_speakers_id(model): if model: - model_data = torch.load(os.path.join(now_dir,model), map_location="cpu") + 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)) @@ -331,6 +331,7 @@ def get_speakers_id(model): else: return [0] + # Inference tab def inference_tab(): default_weight = names[0] if names else None