From a65bf8b90c6d5a7edc24f89bd77fb4fe0100fbb0 Mon Sep 17 00:00:00 2001
From: RyanOnTheInside <7623207+ryanontheinside@users.noreply.github.com>
Date: Tue, 10 Feb 2026 22:57:26 -0500
Subject: [PATCH] fix: text2music instruction passthrough

---
 nodes/acestep/patches.py | 7 +++++++
 pyproject.toml           | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/nodes/acestep/patches.py b/nodes/acestep/patches.py
index ac0e470..4afa800 100644
--- a/nodes/acestep/patches.py
+++ b/nodes/acestep/patches.py
@@ -250,6 +250,13 @@ def _create_patched_tokenizer(original_tokenize_with_weights):
         out = original_tokenize_with_weights(self, text, return_word_ids, **stock_kwargs.copy())
         logger.debug(f"[ACE15_TOKENIZER] generate_audio_codes={out.get('lm_metadata', {}).get('generate_audio_codes', 'N/A')}")
 
+        # For text2music, the stock tokenizer output is correct — don't override.
+        # We only need custom templates for our extended task types.
+        if task_type == "text2music":
+            out["task_type"] = task_type
+            out["track_name"] = track_name
+            return out
+
         # Get appropriate instruction for task type
         instruction = get_task_instruction(task_type, track_name)
 
diff --git a/pyproject.toml b/pyproject.toml
index a82ce37..6777c2b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui_ryanonyheinside"
 description = "Custom nodes introducing everything-reactivity and timelines to ComfyUI"
-version = "2.1.9"
+version = "2.1.10"
 license = {file = "LICENSE"}
 
 dependencies = ["pygame", "opencv-python>=4.5.0", "scipy", "torchaudio", "pillow", "pymunk>=6.8.0,<7.0.0", "matplotlib", "openunmix", "mido", "scikit-image"]