format

vllm-project · Nov 15, 2024 · a68bc4f · a68bc4f
1 parent 1b280b9
commit a68bc4f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/model_executor/models/medusa.py b/vllm/model_executor/models/medusa.py
@@ -19,7 +19,8 @@ def __init__(self, config: VllmConfig, hidden_size: int,
         super().__init__()
 
         self.layers = nn.ModuleList([
-            nn.Linear(hidden_size, hidden_size,
+            nn.Linear(hidden_size,
+                      hidden_size,
                       bias=getattr(config, "medusa_fc_bias", False))
             for _ in range(num_layers)
         ])