fix LiquidAI/LFM2.5-VL-450M chat_template load bug

2026-04-10 23:06:09 +08:00
parent 67f23dd575
commit 3e30360998
13 changed files with 45 additions and 25 deletions
@@ -218,7 +218,7 @@ pub fn load_model<'a>(
            let model = Lfm2GenerateModel::init(path, device, dtype)?;
            ModelInstance::Lfm2(model)
        }
-        WhichModel::LFM2_5VL1_6B | WhichModel::LFM2VL1_6B => {
+        WhichModel::LFM2_5VL1_6B | WhichModel::LFM2VL1_6B | WhichModel::LFM2_5VL450M => {
            let model = Lfm2VLGenerateModel::init(path, device, dtype)?;
            ModelInstance::Lfm2VL(model)
        }