diff --git a/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh b/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh
index d52e175..abbc9d2 100644
--- a/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh
+++ b/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh
@@ -219,7 +219,7 @@ else
 fi
 
 
-GPUS_PER_NODE=2
+GPUS_PER_NODE=1
 NNODES=1
 RANK=0
 MASTER_ENDPOINT=ubuntu