tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Delete local_dir_use_symlinks=False

Browse Source
This commit is contained in:
litagin02
2024-03-13 14:55:42 +09:00
parent 9e5222619f
commit bc89fde15e
1 changed files with 4 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
initialize.py
View File

@@ -16,12 +16,7 @@ def download_bert_models():
for file in v["files"]: for file in v["files"]:
if not Path(local_path).joinpath(file).exists(): if not Path(local_path).joinpath(file).exists():
logger.info(f"Downloading {k} {file}") logger.info(f"Downloading {k} {file}")
hf_hub_download( hf_hub_download(v["repo_id"], file, local_dir=local_path)
v["repo_id"],
file,
local_dir=local_path,
local_dir_use_symlinks=False,
)
def download_slm_model(): def download_slm_model():
@@ -29,12 +24,7 @@ def download_slm_model():
file = "pytorch_model.bin" file = "pytorch_model.bin"
if not Path(local_path).joinpath(file).exists(): if not Path(local_path).joinpath(file).exists():
logger.info(f"Downloading wavlm-base-plus {file}") logger.info(f"Downloading wavlm-base-plus {file}")
hf_hub_download( hf_hub_download("microsoft/wavlm-base-plus", file, local_dir=local_path)
"microsoft/wavlm-base-plus",
file,
local_dir=local_path,
local_dir_use_symlinks=False,
)
def download_pretrained_models(): def download_pretrained_models():
@@ -44,10 +34,7 @@ def download_pretrained_models():
if not Path(local_path).joinpath(file).exists(): if not Path(local_path).joinpath(file).exists():
logger.info(f"Downloading pretrained {file}") logger.info(f"Downloading pretrained {file}")
hf_hub_download( hf_hub_download(
"litagin/Style-Bert-VITS2-1.0-base", "litagin/Style-Bert-VITS2-1.0-base", file, local_dir=local_path
file,
local_dir=local_path,
local_dir_use_symlinks=False,
) )
@@ -58,10 +45,7 @@ def download_jp_extra_pretrained_models():
if not Path(local_path).joinpath(file).exists(): if not Path(local_path).joinpath(file).exists():
logger.info(f"Downloading JP-Extra pretrained {file}") logger.info(f"Downloading JP-Extra pretrained {file}")
hf_hub_download( hf_hub_download(
"litagin/Style-Bert-VITS2-2.0-base-JP-Extra", "litagin/Style-Bert-VITS2-2.0-base-JP-Extra", file, local_dir=local_path
file,
local_dir=local_path,
local_dir_use_symlinks=False,
) )