add LFM2-1.2B, LFM2.5-1.2B-Instruct

2026-03-23 22:07:41 +08:00
parent 77b244e53e
commit 9224593b78
26 changed files with 1001 additions and 40 deletions
@@ -1,8 +1,5 @@
 use aha::models::{
-    deepseek_ocr::config::DeepseekOCRConfig, hunyuan_ocr::config::HunYuanVLConfig,
-    minicpm4::config::MiniCPM4Config, paddleocr_vl::config::PaddleOCRVLConfig,
-    qwen2_5vl::config::Qwen2_5VLConfig, qwen3vl::config::Qwen3VLConfig,
-    voxcpm::config::VoxCPMConfig,
+    deepseek_ocr::config::DeepseekOCRConfig, hunyuan_ocr::config::HunYuanVLConfig, lfm2::config::Lfm2Config, minicpm4::config::MiniCPM4Config, paddleocr_vl::config::PaddleOCRVLConfig, qwen2_5vl::config::Qwen2_5VLConfig, qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig
 };
 use anyhow::Result;

@@ -85,3 +82,16 @@ fn paddleocr_vl_config() -> Result<()> {
    println!("{:?}", config);
    Ok(())
 }
+
+#[test]
+fn lfm2_config() -> Result<()> {
+    // cargo test -F cuda --test config_tests lfm2_config -r -- --nocapture
+    let model_path = "/home/jhq/.aha/LiquidAI/LFM2-1.2B/";
+    // let model_path = "/home/jhq/.aha/LiquidAI/LFM2.5-1.2B-Instruct/";
+    let config_path = model_path.to_string() + "/config.json";
+    let mut config: Lfm2Config = serde_json::from_slice(&std::fs::read(config_path)?)?;
+    println!("{:?}", config);
+    config.full_attn_idx2layer_type();
+    println!("{:?}", config);
+    Ok(())
+}