File size: 625 Bytes
1cf1e13
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from pathlib import Path

from huggingface_hub import hf_hub_download

from config import config


MIRROR: str = config.mirror


def _check_bert(repo_id, files, local_path):
    for file in files:
        if not Path(local_path).joinpath(file).exists():
            if MIRROR.lower() == "openi":
                import openi

                openi.model.download_model(
                    "Stardust_minus/Bert-VITS2", repo_id.split("/")[-1], "./bert"
                )
            else:
                hf_hub_download(
                    repo_id, file, local_dir=local_path, local_dir_use_symlinks=False
                )