Better comments

vllm-project · Oct 16, 2024 · 466ea3e · 466ea3e
1 parent f1cc569
commit 466ea3e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/model_executor/models/llava.py b/vllm/model_executor/models/llava.py
@@ -245,7 +245,8 @@ def __init__(self,
         self.config = config
         self.multimodal_config = multimodal_config
 
-        # HACK: Special cases for pixtral
+        # NOTE: These are special cases for Pixtral-12B in the HF-format
+        # https://huggingface.co/mistral-community/pixtral-12b/blob/main/config.json  # noqa
         if (config.text_config.architectures is None
                 and config.text_config.model_type == "mistral"):
             config.text_config.architectures = ["MistralForCausalLM"]