[SpecV2]: Reopen kl accuracy test for qwen3 + SpecV2 (#22104)
This commit is contained in:
@@ -68,11 +68,10 @@ class TestQwen3NextMTPTopk(
|
|||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
# TODO(hzh): After merging the PR that fixes specv2 to correctly return log probs,
|
class TestQwen3NextMTPV2(GSM8KMixin, KLDivergenceMixin, DefaultServerBase):
|
||||||
# add KLDivergenceMixin back. https://github.com/sgl-project/sglang/pull/18645
|
|
||||||
class TestQwen3NextMTPV2(GSM8KMixin, DefaultServerBase):
|
|
||||||
model = QWEN3_NEXT_MODEL
|
model = QWEN3_NEXT_MODEL
|
||||||
gsm8k_accuracy_thres = 0.93
|
gsm8k_accuracy_thres = 0.93
|
||||||
|
kl_div_thres = 0.0025
|
||||||
other_args = [
|
other_args = [
|
||||||
"--trust-remote-code",
|
"--trust-remote-code",
|
||||||
"--speculative-algorithm",
|
"--speculative-algorithm",
|
||||||
|
|||||||
Reference in New Issue
Block a user