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 = { +