Skip to content

Commit 5ffc2a0

Browse files
committed
use another model
1 parent ac2ea53 commit 5ffc2a0

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

examples/server/tests/unit/test_infill.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,11 +61,10 @@ def test_invalid_input_extra_req(input_extra):
6161
def test_with_qwen_model():
6262
global server
6363
server.model_file = None
64-
server.model_hf_repo = "Qwen/CodeQwen1.5-7B-Chat-GGUF"
65-
server.model_hf_file = "codeqwen-1_5-7b-chat-q2_k.gguf"
64+
server.model_hf_repo = "Qwen/Qwen2.5-Coder-1.5B-Instruct-GGUF"
65+
server.model_hf_file = "qwen2.5-coder-1.5b-instruct-q2_k.gguf"
6666
server.start(timeout_seconds=600)
6767
res = server.make_request("POST", "/infill", data={
68-
"prompt": "Complete this",
6968
"input_extra": [{
7069
"filename": "llama.h",
7170
"text": "LLAMA_API int32_t llama_n_threads();\n"
@@ -74,4 +73,4 @@ def test_with_qwen_model():
7473
"input_suffix": "}\n",
7574
})
7675
assert res.status_code == 200
77-
assert "n_threads" in res.body["content"]
76+
assert res.body["content"].startswith("n_threads")

0 commit comments

Comments
 (0)