[SPEC][5/N] feat: batchsize-aware support for adaptive speculative_num_steps (#24055)
Co-authored-by: 坤钧 <maoyuhan.myh@antgroup.co> Co-authored-by: alphabetc1 <alphabetc1@users.noreply.github.com> Co-authored-by: EanWang211123 <wangyiheng@sangfor.com.cn> Co-authored-by: shuwenn <47200617+alphabetc1@users.noreply.github.com> Co-authored-by: shuwenn <2508695655@qq.com>
This commit is contained in:
co-authored by
坤钧
alphabetc1
EanWang211123
shuwenn
shuwenn
parent
c9f582a272
commit
6b180959a8
@@ -47,11 +47,13 @@ class TestAdaptiveSpeculativeServer(CustomTestCase):
|
||||
with tempfile.NamedTemporaryFile("w", suffix=".json", delete=False) as f:
|
||||
json.dump(
|
||||
{
|
||||
"candidate_steps": [1, 3],
|
||||
"ema_alpha": 1.0,
|
||||
"warmup_batches": 1,
|
||||
"update_interval": 1,
|
||||
"up_hysteresis": 0.0,
|
||||
"1": {
|
||||
"candidate_steps": [1, 3],
|
||||
"ema_alpha": 1.0,
|
||||
"warmup_batches": 1,
|
||||
"update_interval": 1,
|
||||
"up_hysteresis": 0.0,
|
||||
},
|
||||
},
|
||||
f,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user