more fixes

This commit is contained in:
2026-05-14 19:35:39 +00:00
parent 3be051e140
commit 1ceff541b0
3 changed files with 18 additions and 19 deletions

View File

@@ -2214,7 +2214,3 @@ 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)