From 7d16a54f2932e6041c2944972e7e6a09f1654157 Mon Sep 17 00:00:00 2001
From: Tony Salomone <dadmobile@gmail.com>
Date: Mon, 20 Jan 2025 15:53:37 -0500
Subject: [PATCH] Show total model size on DownloadProgressBox

---
 .../components/Shared/DownloadProgressBox.tsx        | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/renderer/components/Shared/DownloadProgressBox.tsx b/src/renderer/components/Shared/DownloadProgressBox.tsx
index 654ea89a..7dd6c612 100644
--- a/src/renderer/components/Shared/DownloadProgressBox.tsx
+++ b/src/renderer/components/Shared/DownloadProgressBox.tsx
@@ -59,6 +59,18 @@ export default function DownloadProgressBox({ jobId, assetName }) {
                           1024
                       )
                     : 'Download Starting'}
+
+                  {downloadProgress?.job_data?.total_size_of_model_in_mb >
+                    0 && (
+                    <>
+                    {'/'}
+                    {formatBytes(
+                      downloadProgress?.job_data?.total_size_of_model_in_mb
+                      * 1024
+                      * 1024
+                    )}
+                    </>
+                  )}
                   ↓
                 </>
               </Typography>
-- 
GitLab