[Feature] Support XiaoMi MIMO Flash V2 (#62)

* [Feature] Support MIMO Flash V2
2025-12-31 10:16:33 +08:00
parent 341dc7f296
commit b3c30a3cb9
12 changed files with 1530 additions and 690 deletions
--- a/vllm_kunlun/models/init.py
+++ b/vllm_kunlun/models/init.py
@@ -82,6 +82,9 @@ def register_model():
        "LlamaForCausalLM",
        "vllm_kunlun.models.llama:LlamaForCausalLM")

-
+    ModelRegistry.register_model(
+        "MiMoV2FlashForCausalLM",
+        "vllm_kunlun.models.mimo_v2_flash:MiMoV2FlashForCausalLM")
+    
 def register_quant_method():
    """to do"""