[Benchmark] Benchmark Running Samples for Qwen3 Omni and Dataset Preparation (#212)

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
Signed-off-by: Zhou Taichang <tzhouam@connect.ust.hk>

examples/offline_inference/qwen3_omni/end2end.py

@@ -170,9 +170,16 @@ def main(args):
     else:
         query_result = query_func()
 
+    if not args.enable_stats:
+        log_file = None
+    else:
+        log_file = os.path.join(args.log_dir, f"omni_llm_pipeline_{args.query_type}")
+
     omni_llm = Omni(
         model=model_name,
         stage_configs_path=args.stage_configs_path,
+        log_file=log_file,
+        log_stats=args.enable_stats,
     )
 
     thinker_sampling_params = SamplingParams(
@@ -364,6 +371,12 @@ def parse_args():
         default=16000,
         help="Sampling rate for audio loading (default: 16000).",
     )
+    parser.add_argument(
+        "--log-dir",
+        type=str,
+        default="logs",
+        help="Log directory (default: logs).",
+    )
     parser.add_argument(
         "--modalities",
         type=str,