squall/picotron
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add tqdm when subprocess

Browse Source
This commit is contained in:
ferdinand.mom 2024-12-18 16:07:34 +00:00
parent f1053e3cbe
commit b49ddac4b4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
create_config.py
View File

@ -68,6 +68,7 @@ def download_hf_model_files(files_to_download, model_name, hf_token, save_dir):
print(f"Downloading {file}...") print(f"Downloading {file}...")
env = os.environ.copy() env = os.environ.copy()
env["HF_HUB_ENABLE_HF_TRANSFER"] = "1" env["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
env["PYTHONUNBUFFERED"] = "1"
result = subprocess.run(model_cmd, shell=True, check=False, env=env) result = subprocess.run(model_cmd, shell=True, check=False, env=env)
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"