Update README.md

README.md

@@ -70,6 +70,7 @@ count_parameters(moe_model)
 Download .py files 
 ```
 from huggingface_hub import HfApi, hf_hub_download
+from tqdm.notebook import tqdm
 import os
 import shutil
 
@@ -84,11 +85,11 @@ files_in_repo = api.list_repo_files(repo_id)
 py_files = [file for file in files_in_repo if file.endswith('.py')]
 
 # Directory to save the downloaded files
-save_dir = "./downloaded_files"
-os.makedirs(save_dir, exist_ok=True)
+save_dir = "."
+#os.makedirs(save_dir, exist_ok=True)
 
 # Download each .py file
-for file_name in py_files:
+for file_name in tqdm(py_files):
     file_path = hf_hub_download(repo_id=repo_id, filename=file_name)
     new_path = os.path.join(save_dir, file_name)
     shutil.move(file_path, new_path)
@@ -102,7 +103,6 @@ Download models that will form the experts, as well as the base model
 ```python
 from Phi_3V_MoE.moe_phi3_v import Phi3VForCausalLMMoE, Phi3VForCausalLMMoEConfig
 
-
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model_name_1 = f"lamm-mit/Cephalo-Phi-3-vision-128k-4b-beta"