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,
|
hidden_states,
|
||||||
cu_seqlens,
|
cu_seqlens,
|
||||||
rotary_pos_emb,
|
rotary_pos_emb,
|
||||||
use_reentrant=(self.config.ckpt_use_reentrant or self.config.ve_ckpt_use_reentrant),
|
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
hidden_states = blk(hidden_states, cu_seqlens=cu_seqlens, rotary_pos_emb=rotary_pos_emb)
|
hidden_states = blk(hidden_states, cu_seqlens=cu_seqlens, rotary_pos_emb=rotary_pos_emb)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user