diff --git a/train.sh b/train.sh
index 6291060..efcc3d6 100755
--- a/train.sh
+++ b/train.sh
@@ -1,9 +1,9 @@
 python train.py \
-    --model_name_or_path decapoda-research/llama-7b-hf \
+    --model_name_or_path $model_path \
     --dataset_name alpaca_data.hf \
     --is_dataset_from_disk True \
     --per_device_train_batch_size 8 \
     --per_device_eval_batch_size 8 \
     --do_train \
     --do_eval \
-    --output_dir test-clm
\ No newline at end of file
+    --output_dir $output_dir
\ No newline at end of file