Upload modeling_dots_vision.py with huggingface_hub

This commit is contained in:
ai-modelscope 2025-08-04 18:49:45 +08:00
parent 9c1b7c2e61
commit 338c0296df

View File

@ -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)