Make model_family parameterised

krai · Jan 15, 2025 · 694c3a7 · 694c3a7
1 parent e13252b
commit 694c3a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/base_llama2_loadgen_experiment/data_axs.json b/base_llama2_loadgen_experiment/data_axs.json
@@ -30,8 +30,9 @@
         [ "__add__", "/open_orca_gpt4_tokenized_llama.sampled_24576.pkl" ]
     ]] ],
 
+    "model_family": "llama2",
     "variant": "7b",
-    "checkpoint_path_query": [ "^^", "substitute", "downloaded,hf_tokeniser,model_family=llama2,variant=#{variant}#" ],
+    "checkpoint_path_query": [ "^^", "substitute", "downloaded,hf_tokeniser,model_family=#{model_family}#,variant=#{variant}#" ],
     "checkpoint_path": [ "^^", "execute", [[
         [ "get_kernel" ],
         [ "byquery", [[ "^^", "get", "checkpoint_path_query" ]] ],