From c10afb44c879218dfa9ef7de9f573d97c61a43c6 Mon Sep 17 00:00:00 2001 From: JonghoLee Date: Thu, 1 Aug 2024 12:10:33 +0000 Subject: [PATCH] Fix matching cases in lora_inference.py --- examples/lora_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/lora_inference.py b/examples/lora_inference.py index 1d3cf9c818947..4fc690776c879 100644 --- a/examples/lora_inference.py +++ b/examples/lora_inference.py @@ -38,7 +38,7 @@ for i, output in enumerate(outputs): prompt = output.prompt generated_text = output.outputs[0].text - matching = expected_output[i].lower() == generated_text.lower() + matching = expected_output[i] == generated_text if not matching: print(f"{i} matching::{matching} Prompt: {prompt!r}, Generated text: {generated_text!r} expected_output: {expected_output[i]!r}") else: