Fix crash when manually generating images

This commit is contained in:
somebody
2022-12-19 20:18:41 -06:00
parent 2c47982f3b
commit 9ce72e7071

View File

@@ -9572,6 +9572,8 @@ def generate_story_image(
# This function is a wrapper around generate_image() that integrates the
# result with the story (read: puts it in the corner of the screen).
log_data = log_data or {}
if not display_prompt:
display_prompt = prompt
koboldai_vars.picture_prompt = display_prompt