From 4dcd5e04c01cbdca5c542f1c827d68d422aa674f Mon Sep 17 00:00:00 2001 From: Anatoliy Talamanov Date: Sat, 4 Jan 2025 09:56:16 +0000 Subject: [PATCH] Update test_llm_pipeline_static.py --- tests/python_tests/test_llm_pipeline_static.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/python_tests/test_llm_pipeline_static.py b/tests/python_tests/test_llm_pipeline_static.py index ca22dab719..d2d3673356 100644 --- a/tests/python_tests/test_llm_pipeline_static.py +++ b/tests/python_tests/test_llm_pipeline_static.py @@ -15,8 +15,8 @@ from common import get_default_properties from common import \ - get_greedy_n_stop_strings_exclude_from_output, \ - get_greedy_n_stop_strings_include_to_output, \ + get_greedy, \ + get_greedy_with_penalties, \ get_multinomial_temperature, \ get_multinomial_all_parameters, \ get_multinomial_temperature_and_presence_penalty, \ @@ -46,8 +46,8 @@ def generate_chat_history(model_path, device, pipeline_config, questions): generation_configs = [ - get_greedy_n_stop_strings_exclude_from_output(), - get_greedy_n_stop_strings_include_to_output() + get_greedy(), + get_greedy_with_penalties() ] @pytest.mark.precommit @pytest.mark.nightly