Skip to content

Commit 5e9064e

Browse files
committed
test: stabilize low-level Qwen sampling check
1 parent 97e3cd8 commit 5e9064e

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

‎tests/test_llama.py‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,9 @@ def test_real_model(llama_cpp_model_path):
118118

119119
output = result[prompt_token_count:]
120120
output_text = model.detokenize(output, special=True)
121-
assert output_text == b"5 times over the"
121+
# Low-level sampling output varies across CPU and Metal backends.
122+
assert len(output) == 4
123+
assert output_text
122124

123125

124126
def test_real_llama(llama_cpp_model_path):

0 commit comments

Comments
 (0)