Fix matching cases in lora_inference.py

HabanaAI · Aug 1, 2024 · c10afb4 · c10afb4
1 parent 6b3a039
commit c10afb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: