diff --git a/hy3dshape/scripts/train_deepspeed.sh b/hy3dshape/scripts/train_deepspeed.sh
index 278cf73..444e5cc 100644
--- a/hy3dshape/scripts/train_deepspeed.sh
+++ b/hy3dshape/scripts/train_deepspeed.sh
@@ -29,10 +29,9 @@ else
     export NCCL_IB_SL=3
     export NCCL_CHECK_DISABLE=1
     export NCCL_P2P_DISABLE=0
-    export NCCL_IB_DISABLE=1
+    export NCCL_IB_DISABLE=0
     export NCCL_LL_THRESHOLD=16384
     export NCCL_IB_CUDA_SUPPORT=1
-    # DELETED: The hardcoded export NCCL_SOCKET_IFNAME and UCX_NET_DEVICES lines
     export NCCL_COLLNET_ENABLE=0
     export SHARP_COLL_ENABLE_SAT=0
     export NCCL_NET_GDR_LEVEL=2