Skip to content

Commit f0c8093

Browse files
Update examples/windows/accuracy_benchmark/perplexity_metrics/perplexity_metrics.py
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: ynankani-nv <[email protected]>
1 parent 6f0ecd3 commit f0c8093

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/windows/accuracy_benchmark/perplexity_metrics/perplexity_metrics.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,8 @@ def perplexity_eval(model_dir, input_len=1024, chunk_size=None):
138138
print(f"[CONFIG] Using provided chunk_size: {effective_chunk_size}")
139139
elif model_cfg_json and "search" in model_cfg_json and "chunk_size" in model_cfg_json["search"]:
140140
# Use chunk_size from existing config file
141-
effective_chunk_size = config["search"]["chunk_size"]
141+
effective_chunk_size = model_cfg_json["search"]["chunk_size"]
142+
kv_chunking_enabled = True
142143
kv_chunking_enabled = True
143144
if DEBUG:
144145
print(f"[CONFIG] Using config file chunk_size: {effective_chunk_size}")

0 commit comments

Comments
 (0)