Skip to content

Commit 2a82381

Browse files
authored
Add llm_gen_config_path to workload settings
1 parent 409d816 commit 2a82381

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

script/add-custom-nvidia-system/dummy_config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,9 @@
66

77
EXPORTS = {
88
C.WorkloadSetting(C.HarnessType.Custom, C.AccuracyTarget(0.99), C.PowerSetting.MaxP): {
9-
9+
llm_fields.llm_gen_config_path: 'code/llama2-70b/tensorrt/generation_config.json',
1010
},
1111
C.WorkloadSetting(C.HarnessType.Custom, C.AccuracyTarget(0.999), C.PowerSetting.MaxP): {
12+
llm_fields.llm_gen_config_path: 'code/llama2-70b/tensorrt/generation_config.json',
1213
},
1314
}

0 commit comments

Comments
 (0)