From a07dd555124b47017edda5ae7165874136cd26a9 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sat, 1 Jun 2024 18:15:11 +0200 Subject: [PATCH] fixed story writer --- lollms/functions/story_writing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/functions/story_writing.py b/lollms/functions/story_writing.py index 10bd0bc..30ac80d 100644 --- a/lollms/functions/story_writing.py +++ b/lollms/functions/story_writing.py @@ -121,7 +121,7 @@ def write_story_section(prompt_ideas: str, llm: Any, story_file_path: str, story f"{start_header_id_template}story_section_writer{end_header_id_template}" ] ) - new_section += llm.fast_gen() + new_section += llm.fast_gen(prompt) # Write the new section to the story file story_path.write_text(new_section)