Skip to content

Commit 85e4d06

Browse files
committed
Shortened output for better consistency
Signed-off-by: Amit Zuker <[email protected]>
1 parent 29fe5ef commit 85e4d06

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/unittest/llmapi/test_llm_pytorch.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -510,8 +510,8 @@ def test_llama_3_1_8b_fp8_with_bf16_lora() -> None:
510510
"美国的首都是哪里?",
511511
]
512512
references = [
513-
"华盛顿特区Washington D.C.)是美国的首都,也是世界上唯一一座不属于任何一州的联邦特区。\n华盛顿特区(",
514-
"华盛顿特区。华盛顿特区是美国的首都和一个行政区。它是由哥伦比亚特区和华盛顿特区的两个行政区",
513+
"华盛顿特区\n华盛顿特区(英文名:Washington, D.C.",
514+
"华盛顿特区。华盛顿特区是美国的首都和一个行政区",
515515
]
516516

517517
lora_config = LoraConfig(lora_dir=[lora_dir],
@@ -531,7 +531,7 @@ def test_llama_3_1_8b_fp8_with_bf16_lora() -> None:
531531

532532
try:
533533
outputs = llm.generate(prompts,
534-
SamplingParams(max_tokens=40),
534+
SamplingParams(max_tokens=20),
535535
lora_request=lora_requests)
536536
assert len(outputs) == len(prompts)
537537
finally:

0 commit comments

Comments
 (0)