fix: 修复 MLA 多个 bug 并缩小测试模型参数

- MLA kv_b_proj 输出维度和 q_rope 切分偏移修复
- 打通 MLA 配置从 ModelConfig 到 DecoderBlock 的传递路径
- rope_theta 配置不再被忽略，MLA 使用 qk_rope_head_dim
- tie_weight 使用 is True 避免 None 隐式生效
- norm_eps/rope base 类型标注修正
- 测试模型参数缩小 (dim=8, head_dim=4)
- 新增 6 种架构配置 × 2 场景的前向传播测试

astrai/config/model_config.py

@@ -106,6 +106,11 @@ class ModelConfig(BaseModelConfig):
     use_qk_norm: Optional[bool] = None
     use_gated_attention: Optional[bool] = None
 
+    # MLA
+    kv_lora_rank: Optional[int] = None
+    qk_nope_head_dim: Optional[int] = None
+    qk_rope_head_dim: Optional[int] = None
+
     # MoE
     ffn_type: str = "mlp"
     n_routed_experts: Optional[int] = None