From f09408a092445a8736c45a86b35b731a0625bfaf Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Fri, 18 Nov 2022 09:27:01 -0500
Subject: [PATCH] Potential fix for
 https://github.com/ebolam/KoboldAI/issues/295

---
 aiserver.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/aiserver.py b/aiserver.py
index 2ef0a38c..de75a0e6 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -97,8 +97,9 @@ from tqdm.auto import tqdm
 old_init = tqdm.__init__
 def new_init(self, *args, **kwargs):
     old_init(self, *args, **kwargs)
-    if(self.ncols == 0 and kwargs.get("ncols") != 0):
-        self.ncols = 99
+    if 'ncols' in kwargs:
+        if(self.ncols == 0 and kwargs.get("ncols") != 0):
+            self.ncols = 99
 tqdm.__init__ = new_init
 
 # Fix some issues with the OPT tokenizer