From 09971e626c34699cc407544c505d2d3756286ab7 Mon Sep 17 00:00:00 2001 From: ochafik Date: Mon, 27 Jan 2025 15:43:03 +0000 Subject: [PATCH] Update test_chat_completion.py --- examples/server/tests/unit/test_chat_completion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/server/tests/unit/test_chat_completion.py b/examples/server/tests/unit/test_chat_completion.py index 1e509f91c..91d629e24 100644 --- a/examples/server/tests/unit/test_chat_completion.py +++ b/examples/server/tests/unit/test_chat_completion.py @@ -325,6 +325,7 @@ def test_completion_without_tool_call(template_name: str, n_predict: int, tools: @pytest.mark.slow +@pytest.mark.parametrize("hf_repo,hf_file,template_override", [ ("lmstudio-community/Meta-Llama-3.1-8B-Instruct-GGUF", "Meta-Llama-3.1-8B-Instruct-Q4_K_M.gguf", None), ("bartowski/gemma-2-2b-it-GGUF", "gemma-2-2b-it-Q4_K_M.gguf", None), ("bartowski/Phi-3.5-mini-instruct-GGUF", "Phi-3.5-mini-instruct-Q4_K_M.gguf", None),