RYDE-WORK/ktransformers
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/ktransformers.git synced 2026-02-06 06:23:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove causal mask

Browse Source
This commit is contained in:
Atream 2025-02-23 07:40:47 +00:00
parent cdb6f896bb
commit 006e8c6abc
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ktransformers/operators/models.py
View File

@ -649,9 +649,12 @@ class KDeepseekV2Model(BaseInjectedModule):
if per_layer_prefill_flag: if per_layer_prefill_flag:
causal_mask = None causal_mask = None
else: else:
causal_mask = self._update_causal_mask( if os.name == 'nt':
attention_mask, inputs_embeds, cache_position, past_key_values, output_attentions causal_mask = self._update_causal_mask(
) attention_mask, inputs_embeds, cache_position, past_key_values, output_attentions
)
else:
causal_mask = None
# embed positions # embed positions
hidden_states = inputs_embeds hidden_states = inputs_embeds