nits

Signed-off-by: Varun Sundar Rabindranath <varun@neuralmagic.com>

tests/lora/test_llama_tp.py

@@ -49,9 +49,10 @@ def do_sample(llm: vllm.LLM, lora_path: str, lora_id: int) -> List[str]:
     # Print the outputs.
     generated_texts: List[str] = []
     for output in outputs:
+        prompt = output.prompt
         generated_text = output.outputs[0].text
         generated_texts.append(generated_text)
-
+        print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
     return generated_texts
 
 
@@ -81,6 +82,7 @@ def v1(run_with_both_engines_lora):
 
 @fork_new_process_for_each_test
 def test_llama_lora(sql_lora_files):
+
     llm = vllm.LLM(MODEL_PATH,
                    enable_lora=True,
                    max_num_seqs=16,