From 0744dea49d43300dbc9c63ca260c2ea29ef76fd5 Mon Sep 17 00:00:00 2001 From: ebolam Date: Tue, 20 Dec 2022 11:41:59 -0500 Subject: [PATCH] Fix for pictures --- koboldai_settings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/koboldai_settings.py b/koboldai_settings.py index f89af9cb..2bcea2f3 100644 --- a/koboldai_settings.py +++ b/koboldai_settings.py @@ -2080,7 +2080,7 @@ class KoboldStoryRegister(object): return None, None filename = os.path.join(self.koboldai_vars.save_paths.generated_images, self.story_settings.prompt_picture_filename) prompt = self.story_settings.prompt_picture_prompt - elif action_id in self.actions: + elif action_id in self.actions and 'picture_filename' in self.actions[action_id]: filename = os.path.join(self.koboldai_vars.save_paths.generated_images, self.actions[action_id]['picture_filename']) prompt = self.actions[action_id]['picture_prompt'] else: