modifying how its being displayed becuase now calculating overall percentage in hf_shard_download

c923ef63dbcdf005f05bdb14d8e52541f4a07c03 · 2024-11-18 13:53:38 -0800 · cadenmackenzie

Files touched

M exo/api/chatgpt_api.py

Diff

commit c923ef63dbcdf005f05bdb14d8e52541f4a07c03
Author: cadenmackenzie <cadenmackenzie@gmail.com>
Date:   Mon Nov 18 13:53:38 2024 -0800

    modifying how its being displayed becuase now calculating overall percentage in hf_shard_download
---
 exo/api/chatgpt_api.py | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/exo/api/chatgpt_api.py b/exo/api/chatgpt_api.py
index e899e816..4656f518 100644
--- a/exo/api/chatgpt_api.py
+++ b/exo/api/chatgpt_api.py
@@ -231,14 +231,10 @@ class ChatGPTAPI:
                         if DEBUG >= 2:
                             print(f"Download status for {model_name}: {status}")
                         
-                        # Calculate overall percentage if we have status
-                        download_percentage = None
-                        if status:
-                            percentages = list(status.values())
-                            if percentages:
-                                download_percentage = sum(percentages) / len(percentages)
-                                if DEBUG >= 2:
-                                    print(f"Calculated download percentage for {model_name}: {download_percentage}")
+                        # Get overall percentage from status
+                        download_percentage = status.get("overall") if status else None
+                        if DEBUG >= 2 and download_percentage is not None:
+                            print(f"Overall download percentage for {model_name}: {download_percentage}")
                         
                         model_pool[model_name] = {
                             "name": pretty,

← 5916defb update setup · back to Exo · adding helper funciton to check file download. also modifyin c61f40c6 →