redmoe-ai-v1 commited on
Commit
42f4178
·
verified ·
1 Parent(s): f984833

Upload modeling_dots_vision.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_dots_vision.py +0 -1
modeling_dots_vision.py CHANGED
@@ -393,7 +393,6 @@ class DotsVisionTransformer(PreTrainedModel):
393
  hidden_states,
394
  cu_seqlens,
395
  rotary_pos_emb,
396
- use_reentrant=(self.config.ckpt_use_reentrant or self.config.ve_ckpt_use_reentrant),
397
  )
398
  else:
399
  hidden_states = blk(hidden_states, cu_seqlens=cu_seqlens, rotary_pos_emb=rotary_pos_emb)
 
393
  hidden_states,
394
  cu_seqlens,
395
  rotary_pos_emb,
 
396
  )
397
  else:
398
  hidden_states = blk(hidden_states, cu_seqlens=cu_seqlens, rotary_pos_emb=rotary_pos_emb)