From 0243bbdbd6301def80a5a873029c182df4d096ae Mon Sep 17 00:00:00 2001
From: aimfordeb <133010426+aimfordeb@users.noreply.github.com>
Date: Mon, 23 Feb 2026 02:53:24 +0200
Subject: [PATCH] Possibly fixed the broken continuity?

---
 funpack.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/funpack.py b/funpack.py
index b40b5832..c36fa53b 100644
--- a/funpack.py
+++ b/funpack.py
@@ -754,6 +754,9 @@ class FunPackStoryWriter:
                     )
 
                 seq_text = llm_tokenizer.decode(generated_ids[0][llm_tokens['input_ids'].shape[1]:], skip_special_tokens=True).strip()
+
+                # Append generated sequence — this is what chains everything
+                messages.append({"role": "assistant", "content": seq_text})
                 
                 if sanity_check == False:
                     print(f"[FunPackStoryWriter] Sequence {seq_idx + 1} (sanity check skipped): {seq_text}...")
@@ -807,9 +810,6 @@ class FunPackStoryWriter:
                     print(f"[FunPackStoryWriter] Sequence {seq_idx + 1} (sanity check performed): {seq_text}...")
                 
                 outputs[seq_idx] = seq_text
-
-                # Append generated sequence — this is what chains everything
-                messages.append({"role": "assistant", "content": seq_text})
                 
             return tuple(outputs)
 
@@ -1479,5 +1479,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
 
 
 
+