diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index b950a578e..c65362594 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -29,7 +29,7 @@ suites = { TestFile("models/test_compressed_tensors_models.py", 42), TestFile("models/test_cross_encoder_models.py", 100), TestFile("models/test_embedding_models.py", 73), - TestFile("models/test_encoder_embedding_models.py", 100), + TestFile("models/test_encoder_embedding_models.py", 460), TestFile("models/test_generation_models.py", 103), TestFile("models/test_nvidia_nemotron_nano_v2.py", 180), TestFile("models/test_qwen_models.py", 82), @@ -108,7 +108,7 @@ suites = { TestFile("test_regex_constrained.py", 64), TestFile("test_request_queue_validation.py", 30), TestFile("test_retract_decode.py", 54), - TestFile("test_score_api.py", 180), + TestFile("test_score_api.py", 310), TestFile("test_server_args.py", 1), TestFile("test_skip_tokenizer_init.py", 117), TestFile("test_srt_endpoint.py", 130),