change mimo v2 pro arch (#986)

Author: JamesBrianD

python/sgl_jax/srt/models/mimo_v2_pro.py

@@ -19,7 +19,7 @@
 logger = logging.getLogger(__name__)
 
 
-class MiMoV2ProForCausalLM(MiMoV2FlashForCausalLM):
+class MiMoV2ForCausalLM(MiMoV2FlashForCausalLM):
 
     def __init__(
         self,
@@ -240,4 +240,4 @@ def _create_layer_mappings(self, layer_idx: int) -> dict:
         return mappings
 
 
-EntryClass = MiMoV2ProForCausalLM
+EntryClass = MiMoV2ForCausalLM