From 0903b849b290b13d1a6ae28c592b3095b772fda8 Mon Sep 17 00:00:00 2001 From: Daniel Maddern Date: Wed, 19 Aug 2026 22:54:32 +0700 Subject: [PATCH] Flatten _text_run_ids output (input_ids is [1,seq]) --- src/h3_blackwell_runtime/qwen3vl_vision.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/h3_blackwell_runtime/qwen3vl_vision.py b/src/h3_blackwell_runtime/qwen3vl_vision.py index 48c557e..cd7e26d 100644 --- a/src/h3_blackwell_runtime/qwen3vl_vision.py +++ b/src/h3_blackwell_runtime/qwen3vl_vision.py @@ -384,7 +384,9 @@ def _text_run_ids(prompt: str) -> list[int]: tokenizer_dir = Path(__file__).with_name("qwen25_tokenizer") if not tokenizer_dir.exists(): raise FileNotFoundError(f"Qwen tokenizer directory missing: {tokenizer_dir}") - return H3PromptTokenizer(tokenizer_dir)(prompt or " ") + ids = H3PromptTokenizer(tokenizer_dir)(prompt or " ") + # input_ids is [1, seq]; flatten to a Python list of ints. + return [int(t) for t in ids.reshape(-1).tolist()] @dataclass