From 3be051e1400fd59a1ce3a906eab23a3f549ec247 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 14 May 2026 19:29:47 +0000 Subject: [PATCH] fix --- vllm/patches/deepseek_v4.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/vllm/patches/deepseek_v4.py b/vllm/patches/deepseek_v4.py index 9de3ef6b..2d76d299 100644 --- a/vllm/patches/deepseek_v4.py +++ b/vllm/patches/deepseek_v4.py @@ -2213,3 +2213,8 @@ class DeepseekV4ForCausalLM(nn.Module): def get_expert_mapping(self) -> list[tuple[str, str, int, str]]: return self.model.get_expert_mapping() + + +# Register model architecture with vLLM +from vllm.model_executor.models import ModelRegistry +ModelRegistry.register_model("DeepseekV4ForCausalLM", DeepseekV4ForCausalLM)