From 4ec22e068cd49c2d8458c758ba20c7ef5b20a934 Mon Sep 17 00:00:00 2001 From: Ilya Lavrenov Date: Fri, 27 Dec 2024 12:47:14 +0100 Subject: [PATCH] Fix 3 --- tests/python_tests/test_continuous_batching.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/python_tests/test_continuous_batching.py b/tests/python_tests/test_continuous_batching.py index 128d8b7413..7837a41c01 100644 --- a/tests/python_tests/test_continuous_batching.py +++ b/tests/python_tests/test_continuous_batching.py @@ -170,7 +170,6 @@ def test_post_oom_health(tmp_path, sampling_config): def get_greedy_seq_len_300() -> GenerationConfig: generation_config = GenerationConfig() - generation_config.num_return_sequences = 3 generation_config.max_new_tokens = 300 return generation_config