mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-27 20:04:35 +00:00
104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
import pytest
|
|
from utils import *
|
|
|
|
# We use a F16 MOE gguf as main model, and q4_0 as draft model
|
|
|
|
server = ServerPreset.stories15m_moe()
|
|
|
|
MODEL_DRAFT_FILE_URL = "https://huggingface.co/ggml-org/models/resolve/main/tinyllamas/stories15M-q4_0.gguf"
|
|
|
|
def create_server():
|
|
global server
|
|
server = ServerPreset.stories15m_moe()
|
|
# download draft model file if needed
|
|
file_name = MODEL_DRAFT_FILE_URL.split('/').pop()
|
|
model_draft_file = f'../../../{file_name}'
|
|
if not os.path.exists(model_draft_file):
|
|
print(f"Downloading {MODEL_DRAFT_FILE_URL} to {model_draft_file}")
|
|
with open(model_draft_file, 'wb') as f:
|
|
f.write(requests.get(MODEL_DRAFT_FILE_URL).content)
|
|
print(f"Done downloading draft model file")
|
|
# set default values
|
|
server.model_draft = model_draft_file
|
|
server.draft_min = 4
|
|
server.draft_max = 8
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def fixture_create_server():
|
|
return create_server()
|
|
|
|
|
|
def test_with_and_without_draft():
|
|
global server
|
|
server.model_draft = None # disable draft model
|
|
server.start()
|
|
res = server.make_request("POST", "/completion", data={
|
|
"prompt": "I believe the meaning of life is",
|
|
"temperature": 0.0,
|
|
"top_k": 1,
|
|
})
|
|
assert res.status_code == 200
|
|
content_no_draft = res.body["content"]
|
|
server.stop()
|
|
|
|
# create new server with draft model
|
|
create_server()
|
|
server.start()
|
|
res = server.make_request("POST", "/completion", data={
|
|
"prompt": "I believe the meaning of life is",
|
|
"temperature": 0.0,
|
|
"top_k": 1,
|
|
})
|
|
assert res.status_code == 200
|
|
content_draft = res.body["content"]
|
|
|
|
assert content_no_draft == content_draft
|
|
|
|
|
|
def test_different_draft_min_draft_max():
|
|
global server
|
|
test_values = [
|
|
(1, 2),
|
|
(1, 4),
|
|
(4, 8),
|
|
(4, 12),
|
|
(8, 16),
|
|
]
|
|
last_content = None
|
|
for draft_min, draft_max in test_values:
|
|
server.stop()
|
|
server.draft_min = draft_min
|
|
server.draft_max = draft_max
|
|
server.start()
|
|
res = server.make_request("POST", "/completion", data={
|
|
"prompt": "I believe the meaning of life is",
|
|
"temperature": 0.0,
|
|
"top_k": 1,
|
|
})
|
|
assert res.status_code == 200
|
|
if last_content is not None:
|
|
assert last_content == res.body["content"]
|
|
last_content = res.body["content"]
|
|
|
|
|
|
@pytest.mark.parametrize("n_slots,n_requests", [
|
|
(1, 2),
|
|
(2, 2),
|
|
])
|
|
def test_multi_requests_parallel(n_slots: int, n_requests: int):
|
|
global server
|
|
server.n_slots = n_slots
|
|
server.start()
|
|
tasks = []
|
|
for _ in range(n_requests):
|
|
tasks.append((server.make_request, ("POST", "/completion", {
|
|
"prompt": "I believe the meaning of life is",
|
|
"temperature": 0.0,
|
|
"top_k": 1,
|
|
})))
|
|
results = parallel_function_calls(tasks)
|
|
for res in results:
|
|
assert res.status_code == 200
|
|
assert match_regex("(wise|kind|owl|answer)+", res.body["content"])
|