format code, edit docs

Signed-off-by: Matthias Vogler <[email protected]>
vllm-project · Dec 24, 2024 · f1bf4fe · f1bf4fe
1 parent cc747a7
commit f1bf4fe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docs/source/models/supported_models.rst b/docs/source/models/supported_models.rst
@@ -658,7 +658,7 @@ Text Generation (``--task generate``)
     - Molmo
     - T + I
     - :code:`allenai/Molmo-7B-D-0924`, :code:`allenai/Molmo-72B-0924`, etc.
-    -
+    - ✅︎
     - ✅︎
     - ✅︎
   * - :code:`NVLM_D_Model`

diff --git a/vllm/model_executor/models/molmo.py b/vllm/model_executor/models/molmo.py
@@ -1134,7 +1134,11 @@ 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
-Original file line number
+Diff line change
@@ Expand Up / @@ -658,7 +658,7 @@ Text Generation (``--task generate``) @@
         - Molmo
         - T + I
         - :code:`allenai/Molmo-7B-D-0924`, :code:`allenai/Molmo-72B-0924`, etc.
-        -
+        - ✅︎
         - ✅︎
         - ✅︎
       * - :code:`NVLM_D_Model`
@@ Expand Down @@