mirror of
https://git.datalinker.icu/deepseek-ai/DeepSeek-V3.git
synced 2025-12-09 04:44:28 +08:00
Compare commits
2 Commits
6bc0d0eb02
...
dc74e9aac3
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
dc74e9aac3 | ||
|
|
3421621d7b |
@ -679,7 +679,7 @@ class MoE(nn.Module):
|
||||
y = torch.zeros_like(x)
|
||||
counts = torch.bincount(indices.flatten(), minlength=self.n_routed_experts).tolist()
|
||||
for i in range(self.experts_start_idx, self.experts_end_idx):
|
||||
if counts[i] == 0:
|
||||
if counts[i] == 0 or self.experts[i] is None:
|
||||
continue
|
||||
expert = self.experts[i]
|
||||
idx, top = torch.where(indices == i)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user