Skip to content

Commit

Permalink
format code, edit docs
Browse files Browse the repository at this point in the history
Signed-off-by: Matthias Vogler <[email protected]>
  • Loading branch information
Matthias Vogler committed Dec 24, 2024
1 parent 1ca7a96 commit 2bb466c
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
2 changes: 1 addition & 1 deletion docs/source/models/supported_models.md
Original file line number Diff line number Diff line change
Expand Up @@ -660,7 +660,7 @@ See [this page](#generative-models) for more information on how to use generativ
- Molmo
- T + I
- :code:`allenai/Molmo-7B-D-0924`, :code:`allenai/Molmo-72B-0924`, etc.
-
- ✅︎
- ✅︎
- ✅︎
* - :code:`NVLM_D_Model`
Expand Down
6 changes: 5 additions & 1 deletion vllm/model_executor/models/molmo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 2bb466c

Please sign in to comment.