From ecf7b056f60e66bd0efb427b1b2aa50b97e3f11e Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Mon, 20 Mar 2023 09:06:24 +0000
Subject: [PATCH] add

---
 projects/gpt/auto_export_gpt_345M_single_card.sh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/projects/gpt/auto_export_gpt_345M_single_card.sh b/projects/gpt/auto_export_gpt_345M_single_card.sh
index 2556af0f5..376d893ef 100644
--- a/projects/gpt/auto_export_gpt_345M_single_card.sh
+++ b/projects/gpt/auto_export_gpt_345M_single_card.sh
@@ -17,14 +17,14 @@
 log_dir=log_345m_mp1
 rm -rf $log_dir
 
-DIRECTORY=./auto_infer
+DIRECTORY=./pretrained
 if [ ! -d "$DIRECTORY" ]; then
   echo "start download ckpt"
-  wget -O https://paddlefleetx.bj.bcebos.com/model/nlp/gpt/GPT_auto_345M.tar.gz
-  tar -zxvf GPT_auto_345M.tar.gz
+  wget https://paddlefleetx.bj.bcebos.com/model/nlp/gpt/GPT_345M_FP16.tar.gz
+  tar -zxvf GPT_345M_FP16.tar.gz
 fi
 
 python -m paddle.distributed.launch --log_dir $log_dir --devices "1" \
     ./tools/auto_export.py \
     -c ./ppfleetx/configs/nlp/gpt/auto/generation_gpt_345M_single_card.yaml \
-    -o Engine.save_load.ckpt_dir=./auto_infer/auto
+    -o Engine.save_load.ckpt_dir=./pretrained/auto