baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Temporarily skip sampling tests.

Browse source
This commit is contained in:
Andrei Betlen 2023-05-01 15:01:49 -04:00
parent dbe0ad86c8
commit 2f8a3adaa4
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_llama.py
View file

@ -1,3 +1,4 @@
import pytest
import llama_cpp import llama_cpp
MODEL = "./vendor/llama.cpp/models/ggml-vocab.bin" MODEL = "./vendor/llama.cpp/models/ggml-vocab.bin"
@ -14,6 +15,7 @@ def test_llama():
assert llama.detokenize(llama.tokenize(text)) == text assert llama.detokenize(llama.tokenize(text)) == text
@pytest.mark.skip(reason="need to update sample mocking")
def test_llama_patch(monkeypatch): def test_llama_patch(monkeypatch):
llama = llama_cpp.Llama(model_path=MODEL, vocab_only=True) llama = llama_cpp.Llama(model_path=MODEL, vocab_only=True)
@ -95,6 +97,7 @@ def test_llama_pickle():
assert llama.detokenize(llama.tokenize(text)) == text assert llama.detokenize(llama.tokenize(text)) == text
@pytest.mark.skip(reason="need to update sample mocking")
def test_utf8(monkeypatch): def test_utf8(monkeypatch):
llama = llama_cpp.Llama(model_path=MODEL, vocab_only=True) llama = llama_cpp.Llama(model_path=MODEL, vocab_only=True)