diff --git a/test/registered/lora/test_lora_hf_sgl_logprob_diff.py b/test/registered/lora/test_lora_hf_sgl_logprob_diff.py index 10c03e68a..ddb671373 100644 --- a/test/registered/lora/test_lora_hf_sgl_logprob_diff.py +++ b/test/registered/lora/test_lora_hf_sgl_logprob_diff.py @@ -38,10 +38,8 @@ from sglang.test.ci.ci_register import register_cuda_ci from sglang.test.runners import HFRunner, SRTRunner register_cuda_ci( - est_time=300, - suite="nightly-1-gpu", - nightly=True, - disabled="Temporarily disabled, will be fixed later", + est_time=150, + suite="stage-b-test-small-1-gpu", ) from sglang.test.test_utils import DEFAULT_PORT_FOR_SRT_TEST_RUNNER, CustomTestCase @@ -56,7 +54,7 @@ LOGPROB_THRESHOLD = 1e-01 DEFAULT_TEST_PROMPTS = [ "SGL is a", "AI is a field of computer science focused on", - "Computer science is the study of", + "Write a long story.", "Write a short story.", "What are the main components of a computer?", ]