vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-15 01:29:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update gate.py

Browse source
This commit is contained in:
Atream 2025-03-20 14:54:01 +08:00 • committed by GitHub
parent 32a91c78c1
commit 633af5d235
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ktransformers/operators/gate.py
View file

@ -215,7 +215,7 @@ class KMoEGateDeepSeekV3(BaseInjectedModule, KMoEGateBase):
self.gate_linear = None self.gate_linear = None
def forward(self, hidden_states) -> torch.Tensor: def forward(self, hidden_states) -> torch.Tensor:
if self.is_windows: if True or self.is_windows:
return self.orig_module.forward(hidden_states) return self.orig_module.forward(hidden_states)
bsz, seq_len, h = hidden_states.shape bsz, seq_len, h = hidden_states.shape