diff --git a/tests/poc/llama2_70b/pretrain_llama2_70b_128k_256die_A2_pack_ptd.sh b/tests/poc/llama31_70b/pretrain_llama31_70b_128k_256die_A2_pack_ptd.sh
similarity index 97%
rename from tests/poc/llama2_70b/pretrain_llama2_70b_128k_256die_A2_pack_ptd.sh
rename to tests/poc/llama31_70b/pretrain_llama31_70b_128k_256die_A2_pack_ptd.sh
index 5a61738c028fd743dc27f20db863a076cfc23d2b..c2e2b43a4c81848cea2d63b93b22ccd5f9e86809 100644
--- a/tests/poc/llama2_70b/pretrain_llama2_70b_128k_256die_A2_pack_ptd.sh
+++ b/tests/poc/llama31_70b/pretrain_llama31_70b_128k_256die_A2_pack_ptd.sh
@@ -124,4 +124,4 @@ torchrun $DISTRIBUTED_ARGS pretrain_gpt.py \
     ${DATA_ARGS} \
     ${CKPT_ARGS} \
     ${OUTPUT_ARGS} \
-    --distributed-backend nccl | tee logs/pretrain_llama2_70b_128k_mcore_256die_A2_pack.log
+    --distributed-backend nccl | tee logs/pretrain_llama31_70b_128k_mcore_256die_A2_pack.log
diff --git a/tests/poc/llama2_70b/pretrain_llama2_70b_128k_256die_A3_pack_ptd.sh b/tests/poc/llama31_70b/pretrain_llama31_70b_128k_256die_A3_pack_ptd.sh
similarity index 97%
rename from tests/poc/llama2_70b/pretrain_llama2_70b_128k_256die_A3_pack_ptd.sh
rename to tests/poc/llama31_70b/pretrain_llama31_70b_128k_256die_A3_pack_ptd.sh
index 706feb3d84250a5a39d4bb3292cd781ba0da7fd0..6ba29a877bad903d32a2d40f326473b0d7945be0 100644
--- a/tests/poc/llama2_70b/pretrain_llama2_70b_128k_256die_A3_pack_ptd.sh
+++ b/tests/poc/llama31_70b/pretrain_llama31_70b_128k_256die_A3_pack_ptd.sh
@@ -131,4 +131,4 @@ torchrun $DISTRIBUTED_ARGS pretrain_gpt.py \
     ${DATA_ARGS} \
     ${CKPT_ARGS} \
     ${OUTPUT_ARGS} \
-    --distributed-backend nccl | tee logs/pretrain_llama2_70b_128k_mcore_256die_A3_pack.log
+    --distributed-backend nccl | tee logs/pretrain_llama31_70b_128k_mcore_256die_A3_pack.log