xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-04-16 14:47:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

update through comments

Browse Source 
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
yewentao256 2025-10-16 22:02:19 +00:00
parent e958c583e7
commit 478b04e423
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/v1/generation/test_batch_invariance.py
View File

@ -59,12 +59,15 @@ def _random_prompt(min_words: int = 1024, max_words: int = 1024 * 2) -> str:
# Pick a random template
base_prompt = random.choice(prompt_templates)
# Add some padding to vary the length if needed
if min_words > 50:
if max_words < min_words:
max_words = min_words
target_words = random.randint(min_words, max_words)
if target_words > 50:
# For longer prompts, repeat context
padding_text = (
" This is an interesting topic that deserves more explanation. "
* (min_words // 50)
* (target_words // 50)
)
base_prompt = base_prompt + padding_text