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

Add fix for status bar when chat mode ends early

Browse Source
This commit is contained in:
ebolam
2022-10-15 18:14:48 -04:00
parent c3f44adea8
commit e738e164e4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
koboldai_settings.py
View File

@@ -675,7 +675,7 @@ class model_settings(settings):
self.tqdm.reset(total=self.genamt) self.tqdm.reset(total=self.genamt)
self.tqdm_progress = 0 self.tqdm_progress = 0
else: else:
self.tqdm.update(1) self.tqdm.update(value-self.tqdm.n)
self.tqdm_progress = int(float(self.generated_tkns)/float(self.genamt)*100) self.tqdm_progress = int(float(self.generated_tkns)/float(self.genamt)*100)
if self.tqdm.format_dict['rate'] is not None: if self.tqdm.format_dict['rate'] is not None:
self.tqdm_rem_time = str(datetime.timedelta(seconds=int(float(self.genamt-self.generated_tkns)/self.tqdm.format_dict['rate']))) self.tqdm_rem_time = str(datetime.timedelta(seconds=int(float(self.genamt-self.generated_tkns)/self.tqdm.format_dict['rate'])))