update model_name

This commit is contained in:
jhqxxx
2026-03-30 20:33:48 +08:00
parent db2d3a1eb4
commit ccbe799b90
15 changed files with 106 additions and 175 deletions
+6 -2
View File
@@ -40,7 +40,11 @@ impl<'a> MiniCPMGenerateModel<'a> {
let model_list = find_type_files(path, "safetensors")?;
let vb = unsafe { VarBuilder::from_mmaped_safetensors(&model_list, dtype, device)? };
let minicpm = MiniCPMModel::new(vb, cfg)?;
let model_name = std::path::Path::new(path)
.file_name()
.and_then(|s| s.to_str())
.unwrap_or("minicpm4")
.to_string();
Ok(MiniCPMGenerateModel {
chat_template,
tokenizer,
@@ -48,7 +52,7 @@ impl<'a> MiniCPMGenerateModel<'a> {
device: device.clone(),
endoftext_id,
im_end_id,
model_name: "minicpm4".to_string(),
model_name,
})
}
}