diff --git a/9g_lora.sh b/9g_lora.sh
index 09f426c..9bb77cd 100644
--- a/9g_lora.sh
+++ b/9g_lora.sh
@@ -46,7 +46,7 @@ OPTS+=" --save-origin-model"
 
 OPTS+=" $@"
 
-CMD="torchrun --nnodes=1 --nproc_per_node=2 --rdzv_id=1 --rdzv_backend=c10d --rdzv_endpoint=${MASTER_ADDR}:${MASTER_PORT} ${CPM_PATH}/apps/cpm9g/sft_cpm9g_delta.py ${OPTS}"
+CMD="FORCE_TORCHRUN=1 CUDA_VISIBLE_DEVICES=1,2,3,4,5,6 torchrun --nnodes=1 --nproc_per_node=2 --rdzv_id=1 --rdzv_backend=c10d --rdzv_endpoint=${MASTER_ADDR}:${MASTER_PORT} ${CPM_PATH}/apps/cpm9g/sft_cpm9g_delta.py ${OPTS}"
 
 echo "${CMD}"
 $CMD
\ No newline at end of file