From b49ddac4b4ef6c9967892368a5fdca0f1da15eaa Mon Sep 17 00:00:00 2001
From: "ferdinand.mom" <ferdinand.mom@epita.fr>
Date: Wed, 18 Dec 2024 16:07:34 +0000
Subject: [PATCH] add tqdm when subprocess

---
 create_config.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/create_config.py b/create_config.py
index 53ef674..a5e15cd 100644
--- 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"