From b910a95c144fc906782d645af4cef51c7416ab4b Mon Sep 17 00:00:00 2001
From: "sungho.park" <sungho.park@webtoonscorp.com>
Date: Thu, 16 Mar 2023 14:06:02 +0000
Subject: [PATCH] add

---
 train.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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