fixed story writer

This commit is contained in:
Saifeddine ALOUI 2024-06-01 18:15:11 +02:00
parent 9a8f65853d
commit a07dd55512

View File

@ -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}" 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 # Write the new section to the story file
story_path.write_text(new_section) story_path.write_text(new_section)