Skip to content

Commit 9756373

Browse files
littlegyCUHKSZzxylvhan028
authored
Add test for "generate" endpoint (#4181)
* TEST: add generate test * update test_input_ids_mode * fix lint * update rl test * fix mm processor args * comment * optimize * add test case * add comments * fix typo * fix generate * warning * remove * check parameters for /generate endpoint * update generate test * update toolkit --------- Co-authored-by: zxy <zhou0493@e.ntu.edu.sg> Co-authored-by: lvhan028 <lvhan_028@163.com>
1 parent c5a05c1 commit 9756373

File tree

5 files changed

+1222
-5
lines changed

5 files changed

+1222
-5
lines changed

autotest/evaluate/test_api_evaluate.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -171,9 +171,9 @@ def get_turbomind_model_list(tp_num):
171171
new_model_list = []
172172
for model in model_list:
173173
if 'Qwen3-235B-A22B-Thinking-2507' in model['model']:
174-
model['extra'] = '--session-len 65536 --cache-max-entry-count 0.9 --max-batch-size 1024 '
174+
model['extra'] += '--session-len 65536 --cache-max-entry-count 0.9 --max-batch-size 1024 '
175175
else:
176-
model['extra'] = '--session-len 65536 --cache-max-entry-count 0.9 '
176+
model['extra'] += '--session-len 65536 --cache-max-entry-count 0.9 '
177177
model['cuda_prefix'] = None
178178
new_model_list.append(model)
179179
return new_model_list
@@ -184,9 +184,9 @@ def get_pytorch_model_list(tp_num):
184184
new_model_list = []
185185
for model in model_list:
186186
if 'Qwen3-235B-A22B-Thinking-2507' in model['model']:
187-
model['extra'] = '--session-len 65536 --cache-max-entry-count 0.9 --max-batch-size 1024 '
187+
model['extra'] += '--session-len 65536 --cache-max-entry-count 0.9 --max-batch-size 1024 '
188188
else:
189-
model['extra'] = '--session-len 65536 --cache-max-entry-count 0.9 '
189+
model['extra'] += '--session-len 65536 --cache-max-entry-count 0.9 '
190190
model['cuda_prefix'] = None
191191
new_model_list.append(model)
192192
return new_model_list

0 commit comments

Comments
 (0)