diff --git a/model/run_hf.py b/model/run_hf.py
index 213dc5c..891c02f 100644
--- 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)