add tqdm when subprocess

2024-12-18 16:07:34 +00:00 · 2024-12-18 16:07:34 +00:00 · b49ddac4b4
commit b49ddac4b4
parent f1053e3cbe
1 changed files with 1 additions and 0 deletions
--- a/create_config.py
+++ b/create_config.py
@ -68,6 +68,7 @@ def download_hf_model_files(files_to_download, model_name, hf_token, save_dir):
        print(f"Downloading {file}...")
        env = os.environ.copy()
        env["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
+        env["PYTHONUNBUFFERED"] = "1"
        result = subprocess.run(model_cmd, shell=True, check=False, env=env)            
        
        os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"