Skip to content

GitLab

Commit b99b5676 authored by zhuwenwen's avatar zhuwenwen
Browse files

fix glm register error

parent b62fe3c1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/config.py
View file @ b99b5676
...@@ -423,7 +423,7 @@ class ModelConfig: ...@@ -423,7 +423,7 @@ class ModelConfig:
self, limit_mm_per_prompt: Optional[Mapping[str, int]] self, limit_mm_per_prompt: Optional[Mapping[str, int]]
) -> Optional["MultiModalConfig"]: ) -> Optional["MultiModalConfig"]:
architectures = getattr(self.hf_config, "architectures", []) architectures = getattr(self.hf_config, "architectures", [])
if ModelRegistry.is_multimodal_model(architectures): if ModelRegistry.is_multimodal_model(architectures) and hasattr(self.hf_config, "vision_config"):
return MultiModalConfig(limit_per_prompt=limit_mm_per_prompt or {}) return MultiModalConfig(limit_per_prompt=limit_mm_per_prompt or {})
if limit_mm_per_prompt: if limit_mm_per_prompt:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment