diff --git a/benchmarks/locomo_bench.py b/benchmarks/locomo_bench.py index dd6dbc4..cdfbb58 100644 --- a/benchmarks/locomo_bench.py +++ b/benchmarks/locomo_bench.py @@ -665,7 +665,7 @@ def run_benchmark( palace_cache = json.load(f) print(f" Palace cache: {len(palace_cache)} room assignments loaded") - rerank_label = f" + LLM re-rank ({llm_model.split('-')[1]})" if llm_rerank_enabled else "" + rerank_label = f" + LLM re-rank ({llm_model})" if llm_rerank_enabled else "" print(f"\n{'=' * 60}") print(" MemPal × LoCoMo Benchmark") diff --git a/benchmarks/longmemeval_bench.py b/benchmarks/longmemeval_bench.py index 2cb9836..1cdc865 100644 --- a/benchmarks/longmemeval_bench.py +++ b/benchmarks/longmemeval_bench.py @@ -3015,7 +3015,7 @@ def run_benchmark( if unique_sessions and api_key and not skip_precompute: print( - f" Diary ingest: pre-computing {len(unique_sessions)} sessions with {llm_model.split('-')[1]}..." + f" Diary ingest: pre-computing {len(unique_sessions)} sessions with {llm_model}..." ) done = 0 cache_path = Path(diary_cache_file) if diary_cache_file else None