Update test_llm_pipeline_static.py

openvinotoolkit · Jan 4, 2025 · 4dcd5e0 · 4dcd5e0
1 parent 5afde89
commit 4dcd5e0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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