Skip to content

Commit

Permalink
Merge pull request #902 from IAHispano/formatter/main
Browse files Browse the repository at this point in the history
chore(format): run black on main
  • Loading branch information
Vidalnt authored Dec 4, 2024
2 parents dbf1e9c + 02d7399 commit e8bed56
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tabs/inference/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand All @@ -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
Expand Down

0 comments on commit e8bed56

Please sign in to comment.