Add a test case to test retract (#1662)

2024-10-13 20:32:37 -07:00
parent 2725f8da61
commit 869f1c02c4
4 changed files with 50 additions and 1 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -17,6 +17,7 @@ suites = {
        "test_large_max_new_tokens.py",
        "test_openai_server.py",
        "test_pytorch_sampling_backend.py",
+        "test_retract_decode.py",
        "test_server_args.py",
        "test_skip_tokenizer_init.py",
        "test_srt_engine.py",