Skip to content

Commit 5d81953

Browse files
committed
1. fix lint
1 parent b8daff7 commit 5d81953

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

lmdeploy/pytorch/models/deepseek_v2.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -577,11 +577,7 @@ def forward(
577577
class MoEGate(nn.Module):
578578
"""Deepseek Gate."""
579579

580-
def __init__(self,
581-
config: Any,
582-
dtype: torch.dtype = None,
583-
device: torch.device = None,
584-
info: Any = None):
580+
def __init__(self, config: Any, dtype: torch.dtype = None, device: torch.device = None, info: Any = None):
585581
super().__init__()
586582
self.config = config
587583
self.top_k = config.num_experts_per_tok

0 commit comments

Comments
 (0)