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"]