diff --git a/examples/text-generation/run_generation.py b/examples/text-generation/run_generation.py
index b4c38574ab..d529dc9ee9 100755
--- a/examples/text-generation/run_generation.py
+++ b/examples/text-generation/run_generation.py
@@ -530,7 +530,7 @@ def main():
         per_sequence_profiler = disabled_profiler
         per_token_profiler = active_profiler
 
-    if args.dataset_name == "mlcommons":
+    if args.dataset_name == "openorca" or args.dataset_name == "mlcommons":
         # Benchmark over the prompts below
         def get_ds(args):
             ds = pd.read_pickle(args.mlcommons_dataset)