Upload modeling_dots_vision.py with huggingface_hub
Browse files- 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)
|