From 037bec88ac3a751a11c443e99bc1cff7bc995452 Mon Sep 17 00:00:00 2001
From: Matthias Vogler <matthias.vogler@joesecurity.org>
Date: Tue, 24 Dec 2024 10:03:17 +0100
Subject: [PATCH] format code

Signed-off-by: Matthias Vogler <matthias.vogler@joesecurity.org>
---
 vllm/model_executor/models/molmo.py | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/vllm/model_executor/models/molmo.py b/vllm/model_executor/models/molmo.py
index 9b9a2310040f2..0eb4a77d31faf 100644
--- a/vllm/model_executor/models/molmo.py
+++ b/vllm/model_executor/models/molmo.py
@@ -15,7 +15,7 @@
 from vllm.attention import Attention, AttentionMetadata
 from vllm.attention.layer import MultiHeadAttention
 from vllm.compilation.decorators import support_torch_compile
-from vllm.config import CacheConfig, VllmConfig, LoRAConfig
+from vllm.config import CacheConfig, LoRAConfig, VllmConfig
 from vllm.distributed import (get_pp_group, get_tensor_model_parallel_rank,
                               get_tensor_model_parallel_world_size,
                               split_tensor_along_last_dim,
@@ -43,7 +43,7 @@
                            SequenceData)
 from vllm.transformers_utils.processor import get_processor
 
-from .interfaces import SupportsMultiModal, SupportsPP, SupportsLoRA
+from .interfaces import SupportsLoRA, SupportsMultiModal, SupportsPP
 from .utils import (AutoWeightsLoader, WeightsMapper, is_pp_missing_parameter,
                     make_empty_intermediate_tensors_factory, make_layers,
                     maybe_prefix, merge_multimodal_embeddings)
@@ -1121,7 +1121,8 @@ def input_processor_for_molmo(ctx: InputContext, inputs: DecoderOnlyInputs):
 @MULTIMODAL_REGISTRY.register_max_image_tokens(get_max_molmo_image_tokens)
 @INPUT_REGISTRY.register_dummy_data(dummy_data_for_molmo)
 @INPUT_REGISTRY.register_input_processor(input_processor_for_molmo)
-class MolmoForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, SupportsLoRA):
+class MolmoForCausalLM(nn.Module, SupportsMultiModal, SupportsPP,
+                       SupportsLoRA):
     packed_modules_mapping = {
         "att_proj": ["att_proj"],
         "attn_out": ["attn_out"],
@@ -1134,11 +1135,12 @@ class MolmoForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, SupportsLoRA):
     ]
     embedding_modules = {}
     embedding_padding_modules = {}
-    def __init__(
-            self, *,
-            vllm_config: VllmConfig,
-            prefix: str = "",
-            lora_config: Optional[LoRAConfig] = None):
+
+    def __init__(self,
+                 *,
+                 vllm_config: VllmConfig,
+                 prefix: str = "",
+                 lora_config: Optional[LoRAConfig] = None):
         super().__init__()
         config = vllm_config.model_config.hf_config
         quant_config = vllm_config.quant_config
@@ -1167,9 +1169,8 @@ def __init__(
 
         self.make_empty_intermediate_tensors = (
             self.model.make_empty_intermediate_tensors)
-        
+
         self.lora_config = lora_config
-        
 
     def _parse_and_validate_image_input(
         self,