Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/torch-cluster.git" did not exist on "65a6c5cf65b64a6e898b2160c2f6df18775b1735"
Unverified Commit cc74499d authored by Ke Bao's avatar Ke Bao Committed by GitHub
Browse files

Fix draft extend ut stability with flush cache (#6979)

parent 0c1f03a2
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
test/srt/test_eagle_infer.py
View file @ cc74499d
......@@ -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",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment