From 75eb6c6cdd6d4a8ab195c86be1d0171ca540c0df Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Wed, 24 Aug 2022 07:42:54 -0400
Subject: [PATCH] Fix for probability bug

---
 koboldai_settings.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/koboldai_settings.py b/koboldai_settings.py
index e90f7cf4..be4daf82 100644
--- a/koboldai_settings.py
+++ b/koboldai_settings.py
@@ -948,7 +948,8 @@ class KoboldStoryRegister(object):
             old_length = self.actions[action_step]["Selected Text Length"]
             if option_number < len(self.actions[action_step]['Options']):
                 self.actions[action_step]["Selected Text"] = self.actions[action_step]['Options'][option_number]['text']
-                self.actions[action_step]["Probabilities"] = self.actions[action_step]['Options'][option_number]['Probabilities']
+                if 'Probabilities' in self.actions[action_step]['Options'][option_number]:
+                    self.actions[action_step]["Probabilities"] = self.actions[action_step]['Options'][option_number]['Probabilities']
                 if self.tokenizer is not None:
                     self.actions[action_step]['Selected Text Length'] = len(self.tokenizer.encode(self.actions[action_step]['Options'][option_number]['text']))
                 else: