Skip to content

GitLab

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

Update gate.py

parent 32a91c78
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ktransformers/operators/gate.py
View file @ 633af5d2
...@@ -215,7 +215,7 @@ class KMoEGateDeepSeekV3(BaseInjectedModule, KMoEGateBase): ...@@ -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
......
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