From c1a96593fdcb7103da6734baab0f5d0c79a9f652 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Fri, 13 Oct 2023 12:32:33 -0400
Subject: [PATCH] Prevent model loading status from going over 100%

---
 modeling/patches.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modeling/patches.py b/modeling/patches.py
index 5664ec07..01db7b20 100644
--- a/modeling/patches.py
+++ b/modeling/patches.py
@@ -220,6 +220,7 @@ class LazyloadPatches:
 # BEGIN PATCH
         utils.bar = tqdm(total=len(state_dict), desc="Loading model tensors", file=utils.UIProgressBarFile(), position=1)
         utils.koboldai_vars.total_layers = len(state_dict)
+        utils.koboldai_vars.loaded_layers = 0
 
         for param_name, param in sorted(
             state_dict.items(),