update model_name
This commit is contained in:
@@ -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,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user