rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Potential fix for https://github.com/ebolam/KoboldAI/issues/295

Browse Source
This commit is contained in:
ebolam
2022-11-18 09:27:01 -05:00
parent 0e1a235e3a
commit f09408a092
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@@ -97,8 +97,9 @@ from tqdm.auto import tqdm
old_init = tqdm.__init__ old_init = tqdm.__init__
def new_init(self, *args, **kwargs): def new_init(self, *args, **kwargs):
old_init(self, *args, **kwargs) old_init(self, *args, **kwargs)
if(self.ncols == 0 and kwargs.get("ncols") != 0): if 'ncols' in kwargs:
self.ncols = 99 if(self.ncols == 0 and kwargs.get("ncols") != 0):
self.ncols = 99
tqdm.__init__ = new_init tqdm.__init__ = new_init
# Fix some issues with the OPT tokenizer # Fix some issues with the OPT tokenizer