Upload modeling_dots_vision.py with huggingface_hub
This commit is contained in:
parent
9c1b7c2e61
commit
338c0296df
@ -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)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user