Skip to content

Commit 6493a14

Browse files
author
ochafik
committed
test w/ beefier qwen 2.5 coder 3b
1 parent 789a3e1 commit 6493a14

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

examples/server/tests/unit/test_tool_call.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -167,9 +167,9 @@ def test_completion_with_required_tool_tiny_slow(template_name: str, tool: dict,
167167
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-1.5B-Instruct-GGUF:Q4_K_M", None),
168168
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-1.5B-Instruct-GGUF:Q4_K_M", "chatml"),
169169
170-
(TEST_TOOL, "success", "bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", None),
171-
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", None),
172-
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", "chatml"),
170+
(TEST_TOOL, "success", "bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", None),
171+
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", None),
172+
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", "chatml"),
173173
174174
(TEST_TOOL, "success", "bartowski/Qwen2.5-7B-Instruct-GGUF:Q4_K_M", None),
175175
(PYTHON_TOOL, "code", "bartowski/Qwen2.5-7B-Instruct-GGUF:Q4_K_M", None),
@@ -308,8 +308,8 @@ def test_completion_without_tool_call_slow(template_name: str, n_predict: int, t
308308
("bartowski/Qwen2.5-1.5B-Instruct-GGUF:Q4_K_M", None),
309309
("bartowski/Qwen2.5-1.5B-Instruct-GGUF:Q4_K_M", "chatml"),
310310
311-
("bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", None),
312-
("bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", "chatml"),
311+
("bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", None),
312+
("bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", "chatml"),
313313
314314
("bartowski/Qwen2.5-7B-Instruct-GGUF:Q4_K_M", None),
315315
("bartowski/Qwen2.5-7B-Instruct-GGUF:Q4_K_M", "chatml"),
@@ -383,8 +383,8 @@ def do_test_weather(server: ServerProcess, **kwargs):
383383
@pytest.mark.slow
384384
@pytest.mark.parametrize("result_override,n_predict,hf_repo,template_override", [
385385
(None, 128, "bartowski/Phi-3.5-mini-instruct-GGUF:Q4_K_M", "chatml"),
386-
(None, 128, "bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", None),
387-
(None, 128, "bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF:Q4_K_M", "chatml"),
386+
(None, 128, "bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", None),
387+
(None, 128, "bartowski/Qwen2.5-Coder-3B-Instruct-GGUF:Q4_K_M", "chatml"),
388388
(None, 128, "bartowski/Qwen2.5-7B-Instruct-GGUF:Q4_K_M", "chatml"),
389389
(None, 128, "bartowski/Hermes-2-Pro-Llama-3-8B-GGUF:Q4_K_M", ("NousResearch/Hermes-2-Pro-Llama-3-8B", "tool_use")),
390390
(None, 128, "bartowski/Hermes-3-Llama-3.1-8B-GGUF:Q4_K_M", ("NousResearch/Hermes-3-Llama-3.1-8B", "tool_use")),

0 commit comments

Comments
 (0)