Skip to content

Commit 561d6f8

Browse files
authored
[CI] Change test input in Gemma LoRA test (#8163)
1 parent d1dec64 commit 561d6f8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/lora/test_gemma.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def do_sample(llm: vllm.LLM, lora_path: str, lora_id: int) -> List[str]:
1313
prompts = [
1414
"Quote: Imagination is",
1515
"Quote: Be yourself;",
16-
"Quote: So many books,",
16+
"Quote: Painting is poetry that is seen rather than felt,",
1717
]
1818
sampling_params = vllm.SamplingParams(temperature=0, max_tokens=32)
1919
outputs = llm.generate(
@@ -41,7 +41,8 @@ def test_gemma_lora(gemma_lora_files):
4141
expected_lora_output = [
4242
"more important than knowledge.\nAuthor: Albert Einstein\n",
4343
"everyone else is already taken.\nAuthor: Oscar Wilde\n",
44-
"so little time\nAuthor: Frank Zappa\n",
44+
"and poetry is painting that is felt rather than seen.\n"
45+
"Author: Leonardo da Vinci\n",
4546
]
4647

4748
output1 = do_sample(llm, gemma_lora_files, lora_id=1)

0 commit comments

Comments
 (0)