diff --git a/modeling_dots_vision.py b/modeling_dots_vision.py
index 62a1bdf..00c5cd8 100644
--- a/modeling_dots_vision.py
+++ b/modeling_dots_vision.py
@@ -393,7 +393,6 @@ class DotsVisionTransformer(PreTrainedModel):
                     hidden_states,
                     cu_seqlens,
                     rotary_pos_emb,
-                    use_reentrant=(self.config.ckpt_use_reentrant or self.config.ve_ckpt_use_reentrant),
                 )
             else:
                 hidden_states = blk(hidden_states, cu_seqlens=cu_seqlens, rotary_pos_emb=rotary_pos_emb)