diff --git a/tests/integration/test_lists/waives.txt b/tests/integration/test_lists/waives.txt index ec98cc98595c..b2638cc2be06 100644 --- a/tests/integration/test_lists/waives.txt +++ b/tests/integration/test_lists/waives.txt @@ -232,7 +232,6 @@ unittest/disaggregated/test_transfer_lifecycle_gpu.py::test_transfer_lifecycle_g unittest/executor/test_rpc.py::TestRpcCorrectness::test_incremental_task_async SKIP (https://nvbugs/5741476) unittest/executor/test_rpc_proxy.py SKIP (https://nvbugs/5605741) unittest/executor/test_rpc_worker.py SKIP (https://nvbugs/5605741) -unittest/llmapi/test_llm.py::test_generate_with_stop_words SKIP (https://nvbugs/6891162) unittest/llmapi/test_llm_args.py::TestKvCacheManagerV2AutoResolution::test_required_v2_refuses_incompatible_disagg[nixl-cpp] SKIP (https://nvbugs/6917693) unittest/llmapi/test_llm_args.py::TestKvCacheManagerV2AutoResolution::test_required_v2_refuses_incompatible_disagg[ucx-python] SKIP (https://nvbugs/6917693) unittest/llmapi/test_llm_kv_cache_events.py::test_llm_api_attention_dp_kv_events SKIP (https://nvbugs/6891163) diff --git a/tests/unittest/llmapi/test_llm.py b/tests/unittest/llmapi/test_llm.py index 184e990f08cb..44f8d82f9207 100644 --- a/tests/unittest/llmapi/test_llm.py +++ b/tests/unittest/llmapi/test_llm.py @@ -448,7 +448,10 @@ def test_generate_with_stop_words(): model=llama_model_path, kv_cache_config=global_kvcache_config, ) - stop_id = llm.tokenizer.encode("N", add_special_tokens=False)[-1] + # The model continues "A B C" with space-separated letters, so the token + # it generates for N is the space-prefixed " N"; with BPE tokenizers + # (e.g. Qwen3) that differs from the token of a bare "N". + stop_id = llm.tokenizer.encode(" N", add_special_tokens=False)[-1] llm_check_output(llm, prompts, ["D E F G H I J K L M"],