Fiz yapf check

Signed-off-by: Flavia Beo <[email protected]>
vllm-project · Oct 25, 2024 · 27a4193 · 27a4193
1 parent 1e8558e
commit 27a4193
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/vllm/model_executor/models/llama.py b/vllm/model_executor/models/llama.py
@@ -271,15 +271,13 @@ def forward(
 @support_torch_compile
 class LlamaModel(nn.Module):
 
-    def __init__(
-        self,
-        config: LlamaConfig,
-        cache_config: Optional[CacheConfig] = None,
-        quant_config: Optional[QuantizationConfig] = None,
-        lora_config: Optional[LoRAConfig] = None,
-        prefix: str = "",
-        **kwargs
-    ) -> None:
+    def __init__(self,
+                 config: LlamaConfig,
+                 cache_config: Optional[CacheConfig] = None,
+                 quant_config: Optional[QuantizationConfig] = None,
+                 lora_config: Optional[LoRAConfig] = None,
+                 prefix: str = "",
+                 **kwargs) -> None:
         super().__init__()
         self.config = config
         self.padding_idx = config.pad_token_id