Revert incorrect shortening of llama2-70b

krai · Jan 28, 2025 · 9775cd8 · 9775cd8
1 parent 5ecf42a
commit 9775cd8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/base_loadgen_program/data_axs.json b/base_loadgen_program/data_axs.json
@@ -141,12 +141,14 @@
         [ "^^", "slice", [ "^^", "get", "param_to_conf_keys" ], { "skip_missing": true } ]
     ],
 
+    "model_name": [ "^^", "substitute", "#{model_family}#-#{variant}#" ],
+
     "internal_to_mlperf_model_name": { "retinanet_openimages": "retinanet", "bert_large": "bert-99" },
     "mlperf_model_name": [ "^^", "execute", [[
 	    [ "get", "internal_to_mlperf_model_name" ],
 	    [ "get", [ [ "^^", "get", "model_name" ], [ "^^", "get", "model_name" ] ] ]
     ]] ],
-    "full_to_shortened_mlperf_model_name": { "bert-99.9": "bert", "bert-99": "bert", "gptj-99.9": "gptj", "gptj-99": "gptj", "llama2-70b": "llama2" },
+    "full_to_shortened_mlperf_model_name": { "bert-99.9": "bert", "bert-99": "bert", "gptj-99.9": "gptj", "gptj-99": "gptj" },
     "shortened_mlperf_model_name": [ "^^", "execute", [[
 	    [ "get", "full_to_shortened_mlperf_model_name" ],
 	    [ "get", [ [ "^^", "get", "mlperf_model_name" ], [ "^^", "get", "mlperf_model_name" ] ] ]