rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Sync and save show budget

This commit is contained in:
Henk 2022-09-21 17:39:56 +02:00
parent a6298ee6df
commit d0664207e8
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
aiserver.py
View File

@ -402,6 +402,7 @@ class vars:
output_streaming = True output_streaming = True
token_stream_queue = TokenStreamQueue() # Queue for the token streaming token_stream_queue = TokenStreamQueue() # Queue for the token streaming
show_probs = False # Whether or not to show token probabilities show_probs = False # Whether or not to show token probabilities
show_budget = False # Whether or not to show token probabilities
configname = None configname = None
utils.vars = vars utils.vars = vars
@ -1072,6 +1073,7 @@ def savesettings():
js["welcome"] = vars.welcome js["welcome"] = vars.welcome
js["output_streaming"] = vars.output_streaming js["output_streaming"] = vars.output_streaming
js["show_probs"] = vars.show_probs js["show_probs"] = vars.show_probs
js["show_budget"] = vars.show_budget
if(vars.seed_specified): if(vars.seed_specified):
js["seed"] = vars.seed js["seed"] = vars.seed
@ -1195,6 +1197,8 @@ def processsettings(js):
vars.output_streaming = js["output_streaming"] vars.output_streaming = js["output_streaming"]
if("show_probs" in js): if("show_probs" in js):
vars.show_probs = js["show_probs"] vars.show_probs = js["show_probs"]
if("show_budget" in js):
vars.show_probs = js["show_budget"]
if("seed" in js): if("seed" in js):
vars.seed = js["seed"] vars.seed = js["seed"]
@ -3986,6 +3990,10 @@ def get_message(msg):
vars.output_streaming = msg['data'] vars.output_streaming = msg['data']
settingschanged() settingschanged()
refresh_settings() refresh_settings()
elif(msg['cmd'] == 'setshowbudget'):
vars.show_budget = msg['data']
settingschanged()
refresh_settings()
elif(msg['cmd'] == 'setshowprobs'): elif(msg['cmd'] == 'setshowprobs'):
vars.show_probs = msg['data'] vars.show_probs = msg['data']
settingschanged() settingschanged()
@ -5577,6 +5585,7 @@ def refresh_settings():
emit('from_server', {'cmd': 'updatefrmtadsnsp', 'data': vars.formatoptns["frmtadsnsp"]}, broadcast=True) emit('from_server', {'cmd': 'updatefrmtadsnsp', 'data': vars.formatoptns["frmtadsnsp"]}, broadcast=True)
emit('from_server', {'cmd': 'updatesingleline', 'data': vars.formatoptns["singleline"]}, broadcast=True) emit('from_server', {'cmd': 'updatesingleline', 'data': vars.formatoptns["singleline"]}, broadcast=True)
emit('from_server', {'cmd': 'updateoutputstreaming', 'data': vars.output_streaming}, broadcast=True) emit('from_server', {'cmd': 'updateoutputstreaming', 'data': vars.output_streaming}, broadcast=True)
emit('from_server', {'cmd': 'updateshowbudget', 'data': vars.show_budget}, broadcast=True)
emit('from_server', {'cmd': 'updateshowprobs', 'data': vars.show_probs}, broadcast=True) emit('from_server', {'cmd': 'updateshowprobs', 'data': vars.show_probs}, broadcast=True)
# Allow toggle events again # Allow toggle events again

3
static/application.js
View File

@ -2708,6 +2708,9 @@ $(document).ready(function(){
} else if(msg.cmd == "updateoutputstreaming") { } else if(msg.cmd == "updateoutputstreaming") {
// Update toggle state // Update toggle state
$("#setoutputstreaming").prop('checked', msg.data).change(); $("#setoutputstreaming").prop('checked', msg.data).change();
} else if(msg.cmd == "updateshowbudget") {
// Update toggle state
$("#setshowbudget").prop('checked', msg.data).change();
} else if(msg.cmd == "updateshowprobs") { } else if(msg.cmd == "updateshowprobs") {
$("#setshowprobs").prop('checked', msg.data).change(); $("#setshowprobs").prop('checked', msg.data).change();