We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 29fe5ef commit 85e4d06Copy full SHA for 85e4d06
tests/unittest/llmapi/test_llm_pytorch.py
@@ -510,8 +510,8 @@ def test_llama_3_1_8b_fp8_with_bf16_lora() -> None:
510
"美国的首都是哪里?",
511
]
512
references = [
513
- "华盛顿特区(Washington D.C.)是美国的首都,也是世界上唯一一座不属于任何一州的联邦特区。\n华盛顿特区(",
514
- "华盛顿特区。华盛顿特区是美国的首都和一个行政区。它是由哥伦比亚特区和华盛顿特区的两个行政区",
+ "华盛顿特区。\n华盛顿特区(英文名:Washington, D.C.",
+ "华盛顿特区。华盛顿特区是美国的首都和一个行政区",
515
516
517
lora_config = LoraConfig(lora_dir=[lora_dir],
@@ -531,7 +531,7 @@ def test_llama_3_1_8b_fp8_with_bf16_lora() -> None:
531
532
try:
533
outputs = llm.generate(prompts,
534
- SamplingParams(max_tokens=40),
+ SamplingParams(max_tokens=20),
535
lora_request=lora_requests)
536
assert len(outputs) == len(prompts)
537
finally:
0 commit comments