Fix for pictures

This commit is contained in:
ebolam
2022-12-20 11:41:59 -05:00
parent ca9397f670
commit 0744dea49d

View File

@@ -2080,7 +2080,7 @@ class KoboldStoryRegister(object):
return None, None return None, None
filename = os.path.join(self.koboldai_vars.save_paths.generated_images, self.story_settings.prompt_picture_filename) filename = os.path.join(self.koboldai_vars.save_paths.generated_images, self.story_settings.prompt_picture_filename)
prompt = self.story_settings.prompt_picture_prompt 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']) filename = os.path.join(self.koboldai_vars.save_paths.generated_images, self.actions[action_id]['picture_filename'])
prompt = self.actions[action_id]['picture_prompt'] prompt = self.actions[action_id]['picture_prompt']
else: else: