diff --git a/tabs/download/download.py b/tabs/download/download.py index 6ca3e91dd..e70a92fcd 100644 --- a/tabs/download/download.py +++ b/tabs/download/download.py @@ -29,9 +29,13 @@ def save_drop_model(dropbox): model_name = format_title(file_name.split(".pth")[0]) else: if "v2" not in dropbox: - model_name = format_title(file_name.split("_nprobe_1_")[1].split("_v1")[0]) + model_name = format_title( + file_name.split("_nprobe_1_")[1].split("_v1")[0] + ) else: - model_name = format_title(file_name.split("_nprobe_1_")[1].split("_v2")[0]) + model_name = format_title( + file_name.split("_nprobe_1_")[1].split("_v2")[0] + ) model_path = os.path.join(now_dir, "logs", model_name) if not os.path.exists(model_path): os.makedirs(model_path)