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

fix

Browse Source
This commit is contained in:
ebolam
2022-07-02 12:36:32 -04:00
parent 54a5e61ec2
commit 8280063621
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
koboldai_settings.py
View File

@@ -243,7 +243,8 @@ class model_settings(settings):
else: else:
self.tqdm.update(1) self.tqdm.update(1)
self.tqdm_progress = int(float(self.generated_tkns)/float(self.genamt)*100) self.tqdm_progress = int(float(self.generated_tkns)/float(self.genamt)*100)
self.tqdm_rem_time = str(datetime.timedelta(seconds=int(float(self.genamt-self.generated_tkns)/self.tqdm.format_dict['rate']))) 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'])))
#Setup TQDP for model loading #Setup TQDP for model loading
if name == "loaded_layers" and 'tqdm' in self.__dict__: if name == "loaded_layers" and 'tqdm' in self.__dict__:
if value == 0: if value == 0:

2
test_aiserver.py
View File

@@ -13,7 +13,7 @@ def client_data():
app = aiserver.app app = aiserver.app
#app.test_client_class = FlaskLoginClient #app.test_client_class = FlaskLoginClient
client_conn = app.test_client() client_conn = app.test_client()
socketio_client = aiserver.socketio.test_client(app, flask_test_client=client_conn) socketio_client = aiserver.socketio.test_client(app, flask_test_client=client_conn, query_string="ui=1")
#Clear out the connection message #Clear out the connection message
response = socketio_client.get_received() response = socketio_client.get_received()
return (client_conn, app, socketio_client) return (client_conn, app, socketio_client)