rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-03-10 08:30:19 +01:00
Code Issues Projects Releases Wiki Activity

Make sure tqdm progress bars display properly in Colab

Browse Source
This commit is contained in:
Gnome Ann 2022-05-13 17:37:45 -04:00
parent 91d3672446
commit f9f1a5f3a9
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
aiserver.py
View File

@ -57,6 +57,16 @@ if lupa.LUA_VERSION[:2] != (5, 4):
print(f"Please install lupa==1.10. You have lupa {lupa.__version__}.", file=sys.stderr)
# Make sure tqdm progress bars display properly in Colab
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
tqdm.__init__ = new_init
#==================================================================#
# Variables & Storage
#==================================================================#