diff --git a/configs/exp_maniskill2_diffusion_policy/maniskill2_model/pretrained_multimae_rgbd.yaml b/configs/exp_maniskill2_diffusion_policy/maniskill2_model/pretrained_multimae_rgbd.yaml
index 29782ee..eff1d74 100644
--- a/configs/exp_maniskill2_diffusion_policy/maniskill2_model/pretrained_multimae_rgbd.yaml
+++ b/configs/exp_maniskill2_diffusion_policy/maniskill2_model/pretrained_multimae_rgbd.yaml
@@ -25,5 +25,5 @@ model:
           type: low_dim
     obs_encoder:
       rgb_model:
-        _target_: src.models.components.img_encoder.multivit.MultiMAEModel
+        _target_: src.models.components.img_encoder.multivit.MultiViTModel
       use_depth: true
\ No newline at end of file
diff --git a/configs/exp_rlbench_act_policy/rlbench_model/pretrained_multimae_rgbd.yaml b/configs/exp_rlbench_act_policy/rlbench_model/pretrained_multimae_rgbd.yaml
index 1c2135d..318e726 100644
--- a/configs/exp_rlbench_act_policy/rlbench_model/pretrained_multimae_rgbd.yaml
+++ b/configs/exp_rlbench_act_policy/rlbench_model/pretrained_multimae_rgbd.yaml
@@ -16,4 +16,4 @@ trainer:
 model:
   policy:
     backbone: 
-      _target_: src.models.components.img_encoder.multivit.MultiMAEModel
+      _target_: src.models.components.img_encoder.multivit.MultiViTModel
diff --git a/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_multimae_rgbd.yaml b/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_multimae_rgbd.yaml
index f903a5c..c4ce9d5 100644
--- a/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_multimae_rgbd.yaml
+++ b/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_multimae_rgbd.yaml
@@ -29,7 +29,6 @@ model:
           type: low_dim
     obs_encoder:
       rgb_model:
-        _target_: src.models.components.img_encoder.multivit.MultiMAEModel
+        _target_: src.models.components.img_encoder.multivit.MultiViTModel
         type: MultiViT
-        ckpt_path: /mnt/lustre/hetong/.multivit/multimae-b_98_rgb+-depth-semseg_1600e_multivit-afff3f8c.pth
       use_depth: true
\ No newline at end of file
diff --git a/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_ponderv2_pcd.yaml b/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_ponderv2_pcd.yaml
index 64729a1..96c0d16 100644
--- a/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_ponderv2_pcd.yaml
+++ b/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_ponderv2_pcd.yaml
@@ -29,7 +29,7 @@ model:
         _target_: src.models.components.pcd_encoder.spunet.SpUNet
         in_channels: 6
         num_classes: 0
-        pretrained_path: /mnt/lustre/share_data/ponder/pointcept/ponderv2-ppt-pretrain-scannet-s3dis-structured3d.pth
+        pretrained_path: checkpoints/ponderv2-ppt-pretrain-scannet-s3dis-structured3d.pth
       pcd_nsample: 16
       pcd_npoints: 2048
       use_mask: false