minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Tiny cleanup the models' name in test_utils (#15920)

Browse Source
This commit is contained in:
Liangsheng Yin
2025-12-27 14:13:23 +08:00
committed by GitHub
parent 29ce7b3612
commit 9ad546d7e8
15 changed files with 88 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/spec/test_ngram_speculative_decoding.py
+2 -2
View File
@@ -5,7 +5,7 @@ from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.kits.gsm8k_accuracy_kit import GSM8KMixin
from sglang.test.test_utils import (
DEFAULT_NGRAM_SPECULATIVE_TARGET_MODEL_FOR_TEST,
DEFAULT_TARGET_MODEL_NGRAM,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
DEFAULT_URL_FOR_TEST,
CustomTestCase,
@@ -32,7 +32,7 @@ DEFAULT_SERVER_ARGS = [
class TestNgramSpeculativeDecodingBase(GSM8KMixin, CustomTestCase):
model = DEFAULT_NGRAM_SPECULATIVE_TARGET_MODEL_FOR_TEST
model = DEFAULT_TARGET_MODEL_NGRAM
base_url = DEFAULT_URL_FOR_TEST
gsm8k_accuracy_thres = 0.79 # derived tests need to override this
gsm8k_accept_length_thres = 1.8 # derived spec decoding tests need to override this