diff --git a/test/srt/test_eagle_infer.py b/test/srt/test_eagle_infer.py index 1caf447ec..f6dc2cf1f 100644 --- a/test/srt/test_eagle_infer.py +++ b/test/srt/test_eagle_infer.py @@ -612,6 +612,9 @@ class TestEAGLEDraftExtend(CustomTestCase): kill_process_tree(cls.process.pid) def test_one_batch_accept_length(self): + resp = requests.get(self.base_url + "/flush_cache") + self.assertEqual(resp.status_code, 200) + prompts = [ "Hello, my name is", "The president of the United States is",