Skip to content

Commit 6d3cfc3

Browse files
committed
Fixed test_llama.py
1 parent 79f31a9 commit 6d3cfc3

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_llama.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,14 @@ def test_llama_cpp_tokenization():
3636
assert tokens[0] == llama.token_bos()
3737
assert tokens == [1, 15043, 2787]
3838
detokenized = llama.detokenize(tokens)
39-
assert detokenized == text
39+
assert detokenized[1:] == text
4040

4141
tokens = llama.tokenize(text, add_bos=False)
4242
assert tokens[0] != llama.token_bos()
4343
assert tokens == [15043, 2787]
4444

4545
detokenized = llama.detokenize(tokens)
46-
assert detokenized != text
46+
assert detokenized == text
4747

4848
text = b"Hello World</s>"
4949
tokens = llama.tokenize(text)

0 commit comments

Comments
 (0)