From 20eb5eccae6ce433d472df927f8931d3c10ac7cb Mon Sep 17 00:00:00 2001
From: NamH <NamNh0122@gmail.com>
Date: Sun, 18 Feb 2024 15:14:48 +0700
Subject: [PATCH] fix: infinity showed when haven't get total size (#2066)

Signed-off-by: James <james@jan.ai>
Co-authored-by: James <james@jan.ai>
---
 web/hooks/useDownloadState.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/web/hooks/useDownloadState.ts b/web/hooks/useDownloadState.ts
index 3fdd2cbc6..06de9bef6 100644
--- a/web/hooks/useDownloadState.ts
+++ b/web/hooks/useDownloadState.ts
@@ -130,8 +130,8 @@ export const setDownloadStateAtom = atom(
         0
       )
       modelDownloadState.size.transferred = transferredSize
-      modelDownloadState.percent = transferredSize / parentTotalSize
-
+      modelDownloadState.percent =
+        parentTotalSize === 0 ? 0 : transferredSize / parentTotalSize
       currentState[state.modelId] = modelDownloadState
     }