From 36a6e6766d516e2364bfb46f42c9697566ded7cf Mon Sep 17 00:00:00 2001 From: ebolam Date: Thu, 20 Oct 2022 09:59:55 -0400 Subject: [PATCH] Bug Fix --- koboldai_settings.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/koboldai_settings.py b/koboldai_settings.py index d85ab2ed..7717a348 100644 --- a/koboldai_settings.py +++ b/koboldai_settings.py @@ -277,7 +277,7 @@ class koboldai_vars(object): prompt_data = [] for item in reversed(action_text_split): if -1 in item[1]: - tokenized_data = self.tokenizer.encode(item[0]) + tokenized_data = [[x, self.tokenizer.decode(x)] for x in self.tokenizer.encode(item[0])] item[2] = len(tokenized_data) if prompt_length + item[2] <= self.max_prompt_length: prompt_length += item[2] @@ -1526,7 +1526,7 @@ class KoboldStoryRegister(object): #process_variable_changes(self.socketio, "actions", "Options", {"id": self.action_count+1, "options": self.actions[self.action_count+1]["Options"]}, {"id": self.action_count+1, "options": None}) process_variable_changes(self.socketio, "story", 'actions', {"id": self.action_count+1, 'action': self.actions[self.action_count+1]}, None) - def set_probabilites(self, probabilities, action_id=None): + def set_probabilities(self, probabilities, action_id=None): if action_id is None: action_id = self.action_count+1 if action_id in self.actions: