diff --git a/funpack.py b/funpack.py index f4135187..dad82ca3 100644 --- a/funpack.py +++ b/funpack.py @@ -700,7 +700,7 @@ class FunPackStoryWriter: messages = [ {"role": "system", "content": sequence_system_prompt}, {"role": "user", "content": f"""Current sequence number: {seq_idx + 1}\nTotal sequences requested: {prompt_count}\nOriginal user prompt: {user_prompt}n\Generate the next sequence now."""} - {"role": "assistant", "content": f"""Previous sequences for continuity:\n{chr(10).join(outputs[:seq_idx]) if seq_idx > 0 else "This is the first sequence."\n"""} + {"role": "assistant", "content": f"""Previous sequences for continuity:{chr(10).join([f"Sequence {i+1}: {text}" for i, text in enumerate(outputs[:seq_idx])]) if seq_idx > 0 else "This is the first sequence."}"""} ] llm_tokens = llm_tokenizer.apply_chat_template( @@ -1454,5 +1454,6 @@ NODE_DISPLAY_NAME_MAPPINGS = { +