Skip to content

GitLab

Unverified Commit 561d6f80 authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

[CI] Change test input in Gemma LoRA test (#8163)

parent d1dec642
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tests/lora/test_gemma.py
View file @ 561d6f80
...@@ -13,7 +13,7 @@ def do_sample(llm: vllm.LLM, lora_path: str, lora_id: int) -> List[str]: ...@@ -13,7 +13,7 @@ def do_sample(llm: vllm.LLM, lora_path: str, lora_id: int) -> List[str]:
prompts = [ prompts = [
"Quote: Imagination is", "Quote: Imagination is",
"Quote: Be yourself;", "Quote: Be yourself;",
"Quote: So many books,", "Quote: Painting is poetry that is seen rather than felt,",
] ]
sampling_params = vllm.SamplingParams(temperature=0, max_tokens=32) sampling_params = vllm.SamplingParams(temperature=0, max_tokens=32)
outputs = llm.generate( outputs = llm.generate(
...@@ -41,7 +41,8 @@ def test_gemma_lora(gemma_lora_files): ...@@ -41,7 +41,8 @@ def test_gemma_lora(gemma_lora_files):
expected_lora_output = [ expected_lora_output = [
"more important than knowledge.\nAuthor: Albert Einstein\n", "more important than knowledge.\nAuthor: Albert Einstein\n",
"everyone else is already taken.\nAuthor: Oscar Wilde\n", "everyone else is already taken.\nAuthor: Oscar Wilde\n",
"so little time\nAuthor: Frank Zappa\n", "and poetry is painting that is felt rather than seen.\n"
"Author: Leonardo da Vinci\n",
] ]
output1 = do_sample(llm, gemma_lora_files, lora_id=1) output1 = do_sample(llm, gemma_lora_files, lora_id=1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment