format

vllm-project · Nov 15, 2024 · 1b280b9 · 1b280b9
1 parent d596d9b
commit 1b280b9
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
@@ -51,7 +51,8 @@ def __init__(self, *, vllm_config: VllmConfig, prefix: str = "") -> None:
         super().__init__()
         self.config = config
         self.blocks = nn.ModuleList([
-            ResidualBlock(config=config, hidden_size=self.config.hidden_size,
+            ResidualBlock(config=config,
+                          hidden_size=self.config.hidden_size,
                           num_layers=self.config.num_hidden_layers)
             for _ in range(self.config.num_heads)
         ])