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 for probabilities code

Browse Source
This commit is contained in:
ebolam
2022-12-16 19:50:13 -05:00
parent 248994a610
commit 72c0a54f4e
2 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
koboldai_settings.py
View File

@@ -1442,12 +1442,13 @@ class KoboldStoryRegister(object):
if self.actions[i]["Selected Text"] != text:
self.actions[i]["Selected Text"] = text
tokens = self.koboldai_vars.tokenizer.encode(text)
for token_num in range(len(self.actions[i]["Probabilities"])):
for token_option in range(len(self.actions[i]["Probabilities"][token_num])):
if token_num < len(tokens):
self.actions[i]["Probabilities"][token_num][token_option]["Used"] = tokens[token_num] == self.actions[i]["Probabilities"][token_num][token_option]["tokenId"]
else:
self.actions[i]["Probabilities"][token_num][token_option]["Used"] = False
if 'oribabilities' in self.actions[i]:
for token_num in range(len(self.actions[i]["Probabilities"])):
for token_option in range(len(self.actions[i]["Probabilities"][token_num])):
if token_num < len(tokens):
self.actions[i]["Probabilities"][token_num][token_option]["Used"] = tokens[token_num] == self.actions[i]["Probabilities"][token_num][token_option]["tokenId"]
else:
self.actions[i]["Probabilities"][token_num][token_option]["Used"] = False
if "Options" in self.actions[i]:
for j in range(len(self.actions[i]["Options"])):
if self.actions[i]["Options"][j]["text"] == text:
@@ -1640,7 +1641,8 @@ class KoboldStoryRegister(object):
for old_item in old_options:
if item['text'] == old_item['text']:
#We already have this option, so we need to save the probabilities
item['Probabilities'] = old_item['Probabilities']
if 'Probabilities' in old_item:
item['Probabilities'] = old_item['Probabilities']
self.actions[action_id]["Options"].append(item)
process_variable_changes(self.socketio, "story", 'actions', {"id": action_id, 'action': self.actions[action_id]}, None)

12
templates/survey.html
View File

@@ -18,32 +18,32 @@
<div id="choices-container">
<div answer="A" class="choice">
<input type="radio" name="answer">
<input type="radio" name="answer" autocomplete="off">
<span class="choice-text">{{ a }}</span>
</div>
<div answer="B" class="choice">
<input type="radio" name="answer">
<input type="radio" name="answer" autocomplete="off">
<span class="choice-text">{{ b }}</span>
</div>
<div answer="C" class="choice meta-choice">
<input type="radio" name="answer">
<input type="radio" name="answer" autocomplete="off">
<span class="choice-text">Both choices are about the same</span>
</div>
<div answer="D" class="choice meta-choice">
<input type="radio" name="answer">
<input type="radio" name="answer" autocomplete="off">
<span class="choice-text">The first choice is incoherent</span>
</div>
<div answer="E" class="choice meta-choice">
<input type="radio" name="answer">
<input type="radio" name="answer" autocomplete="off">
<span class="choice-text">The second choice is incoherent</span>
</div>
<div answer="F" class="choice meta-choice">
<input type="radio" name="answer">
<input type="radio" name="answer" autocomplete="off">
<span class="choice-text">Both choices are incoherent</span>
</div>
</div>