# pip install huggingface-cli
import os

os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

model_list = [
    "openai-community/gpt2",
    "CompVis/stable-diffusion-v1-4"
]

os.system("pip install -U huggingface-hub")

for model_path in model_list:
    root_dir = os.path.basename(model_path)
    os.system(
        f"huggingface-cli download --resume-download  {model_path} --local-dir ./{root_dir}/{model_path} \
        --local-dir-use-symlinks False")