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

This commit is contained in:
jhqxxx
2026-04-10 23:06:09 +08:00
parent 67f23dd575
commit 3e30360998
13 changed files with 45 additions and 25 deletions
+4 -1
View File
@@ -10,6 +10,8 @@ pub enum WhichModel {
LFM2_5_1_2BInstruct,
#[value(name = "LiquidAI/LFM2.5-VL-1.6B")]
LFM2_5VL1_6B,
#[value(name = "LiquidAI/LFM2.5-VL-450M")]
LFM2_5VL450M,
#[value(name = "LiquidAI/LFM2-VL-1.6B")]
LFM2VL1_6B,
#[value(name = "OpenBMB/MiniCPM4-0.5B")]
@@ -152,7 +154,8 @@ impl WhichModel {
| WhichModel::Qwen3_5_9B
| WhichModel::Qwen3_5Gguf
| WhichModel::LFM2_5VL1_6B
| WhichModel::LFM2VL1_6B => "vlm",
| WhichModel::LFM2VL1_6B
| WhichModel::LFM2_5VL450M => "vlm",
// OCR models
WhichModel::DeepSeekOCR
| WhichModel::DeepSeekOCR2
+1 -1
View File
@@ -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)
}