lint fix

vllm-project · Sep 11, 2024 · 1931480 · 1931480
1 parent 7fb9e90
commit 1931480
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/vllm/model_executor/models/gemma2.py b/vllm/model_executor/models/gemma2.py
@@ -46,6 +46,7 @@
 
 
 class Gemma2MLP(nn.Module):
+
     def __init__(
         self,
         hidden_size: int,
@@ -78,6 +79,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
 
 class Gemma2Attention(nn.Module):
+
     def __init__(self,
                  layer_idx: int,
                  config: Gemma2Config,
@@ -166,6 +168,7 @@ def forward(
 
 
 class Gemma2DecoderLayer(nn.Module):
+
     def __init__(
         self,
         layer_idx: int,
@@ -235,6 +238,7 @@ def forward(
 
 
 class Gemma2Model(nn.Module):
+
     def __init__(
         self,
         config: Gemma2Config,