added missing args

raishish · Aug 12, 2024 · 2bcde55 · 2bcde55
1 parent 132ebb4
commit 2bcde55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/run_hf.py b/model/run_hf.py
@@ -265,6 +265,6 @@ def run_inference(dataset_names, dataset_split, hf_dataset_name, prompt_types, m
     pipe, tokenizer = load_model(hf_model_name, load_in_4bit, load_in_8bit, cache_dir)
 
     # run inference
-    run_inference(dataset_names, dataset_split, hf_dataset_name, prompt_types, model_name, half, use_20_fewshot, pipe, tokenizer)
+    run_inference(dataset_names, dataset_split, hf_dataset_name, prompt_types, model_name, half, use_20_fewshot, pipe, tokenizer, prompt_dir, res_dir)