Unverified Commit ddd35d5b authored by Atream's avatar Atream Committed by GitHub
Browse files

Merge pull request #940 from kvcache-ai/Atream-patch-6

Update gate.py
parents 32a91c78 633af5d2
......@@ -215,7 +215,7 @@ class KMoEGateDeepSeekV3(BaseInjectedModule, KMoEGateBase):
self.gate_linear = None
def forward(self, hidden_states) -> torch.Tensor:
if self.is_windows:
if True or self.is_windows:
return self.orig_module.forward(hidden_states)
bsz, seq_len, h = hidden_states.shape
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment