more fixes
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user