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

Colab fix and send_to_ui fix

Browse Source
This commit is contained in:
ebolam
2022-06-29 17:44:22 -04:00
parent 3bf0d387e0
commit 72827ed149
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@@ -4213,7 +4213,7 @@ def tpumtjgenerate(txt, minimum, maximum, found_entries=None):
koboldai_vars.actions.append_options([x["generated_text"] for x in genout])
genout = [{"generated_text": x['text']} for x in koboldai_vars.actions.get_current_options()]
if(len(koboldai_vars.actions.get_current_options()) == 1):
genresult(koboldai_vars.actions.get_current_options()[0])
genresult(koboldai_vars.actions.get_current_options()[0]['text'])
else:
if(koboldai_vars.lua_koboldbridge.restart_sequence is not None and koboldai_vars.lua_koboldbridge.restart_sequence > 0):
genresult(genout[koboldai_vars.lua_koboldbridge.restart_sequence-1]["generated_text"])

9
koboldai_settings.py
View File

@@ -88,10 +88,11 @@ class koboldai_vars(object):
self._model_settings.send_to_ui()
self._user_settings.send_to_ui()
self._system_settings.send_to_ui()
if 'story' in self._sessions:
self._story_settings[self._sessions['story']].send_to_ui()
else:
self._story_settings['default'].send_to_ui()
#if 'story' in self._sessions:
# self._story_settings[self._sessions['story']].send_to_ui()
#else:
# self._story_settings['default'].send_to_ui()
self._story_settings['default'].send_to_ui()
def __setattr__(self, name, value):
if name[0] == "_":

3
static/koboldai.js
View File

@@ -8,7 +8,7 @@ socket.on("disconnect", (reason, details) => {
});
socket.on('reset_story', function(){reset_story();});
socket.on('var_changed', function(data){var_changed(data);});
socket.onAny(function(event_name, data) {console.log({"event": event_name, "class": data.classname, "data": data});});
//socket.onAny(function(event_name, data) {console.log({"event": event_name, "class": data.classname, "data": data});});
var backend_vars = {};
var presets = {}
@@ -141,6 +141,7 @@ function create_options(data) {
}
function do_story_text_updates(data) {
console.log(data);
story_area = document.getElementById('Selected Text');
if (document.getElementById('Selected Text Chunk '+data.value.id)) {
document.getElementById('Selected Text Chunk '+data.value.id).textContent = data.value.text;