Skip to content

Commit d5b5872

Browse files
committed
Added a way to pass a hf token to the accuarcy script of Llama3.1-8b.
1 parent f6b2b13 commit d5b5872

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

base_small_llm_loadgen_experiment/data_axs.json

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,8 @@
4242

4343
"accuracy_log_dtype": "int32",
4444

45+
"hf_token": "",
46+
4547
"extract_accuracy_report": [ "^^", "execute", [[
4648
[ "get_kernel" ],
4749
[ "byname", "python_script" ],
@@ -59,7 +61,8 @@
5961
"abs_script_path": ["^^", "get", "abs_script_path"],
6062
"script_extra_params": [ "^^", "substitute", "--mlperf-accuracy-file #{accuracy_log_path}# --dataset-file #{dataset_path}# --dtype #{accuracy_log_dtype}#" ],
6163
"desired_python_version": ["^", "kernel_python_major_dot_minor"],
62-
"capture_output": true
64+
"capture_output": true,
65+
"extra_env": { "HUGGING_FACE_HUB_TOKEN": [ "^^", "substitute", "#{hf_token}#" ] }
6366
} ],
6467
0,
6568
[ "func", [ "ufun.rematch", "(\\{.*\\})" ] ],

0 commit comments

Comments
 (0)