Skip to content

Commit

Permalink
Merge pull request IAHispano#897 from IAHispano/formatter/main
Browse files Browse the repository at this point in the history
chore(format): run black on main
  • Loading branch information
blaisewf authored Dec 2, 2024
2 parents f9d04c8 + f71c97a commit 9ec21c2
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
1 change: 0 additions & 1 deletion rvc/lib/tools/model_download.py
Original file line number Diff line number Diff line change
Expand Up @@ -374,7 +374,6 @@ def model_download_pipeline(url: str):
model_name + ".index",
),
)


if success:
print(f"Model {model_name} downloaded!")
Expand Down
8 changes: 6 additions & 2 deletions tabs/download/download.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,9 +38,13 @@ def save_drop_model(dropbox):
model_name = file_name.split(".pth")[0]
elif ".index" in dropbox:
if "v1" in dropbox:
model_name = file_name.split("_nprobe_1_")[1].split("_v1")[0].split(".index")[0]
model_name = (
file_name.split("_nprobe_1_")[1].split("_v1")[0].split(".index")[0]
)
elif "v2" in dropbox:
model_name = file_name.split("_nprobe_1_")[1].split("_v2")[0].split(".index")[0]
model_name = (
file_name.split("_nprobe_1_")[1].split("_v2")[0].split(".index")[0]
)
else:
model_name = file_name.split(".index")[0]
model_path = os.path.join(now_dir, "logs", model_name)
Expand Down

0 comments on commit 9ec21c2

Please sign in to comment.