Rephrase one prompt that generated weirdly formatted output.

HabanaAI · Sep 16, 2024 · 5864c3a · 5864c3a
1 parent 88efc02
commit 5864c3a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/offline_inference_fakehpu.py b/examples/offline_inference_fakehpu.py
@@ -9,6 +9,7 @@
 # Sample prompts.
 prompts = [
     "Berlin is the capital city of ",
+    "Louvre is located in the city of ",
     "Barack Obama was the 44th president of ",
     "Warsaw is the capital city of ",
     "Gniezno is a city in ",
@@ -17,7 +18,7 @@
     "Llanfairpwllgwyngyll is located in country of ",
 ]
 ref_answers = [
-    "Germany", "United States", "Poland", "Poland", "Israel",
+    "Germany", "Paris", "United States", "Poland", "Poland", "Israel",
     "California", "Wales"
 ]
 # Create a sampling params object.