fix error

This commit is contained in:
wxg
2025-09-25 12:25:24 +08:00
parent 4fdab3e7cd
commit b28b6aa25d
2 changed files with 8 additions and 6 deletions
+3 -3
View File
@@ -18,15 +18,15 @@ impl ModelType {
model_path: &str,
device: Option<&Device>,
dtype: Option<DType>,
) -> Result<Box<impl GenerateModel>> {
) -> Result<Box<dyn GenerateModel>> {
match model_type {
ModelType::Qwen2_5VL => {
let model = Qwen2_5VLGenerateModel::init(model_path, device, dtype)?;
Ok(Box::new(model))
Ok(Box::new(model) as Box<dyn GenerateModel>)
},
ModelType::MiniCPM4 => {
let model = MiniCPMGenerateModel::init(model_path, device, dtype)?;
Ok(Box::new(model))
Ok(Box::new(model)as Box<dyn GenerateModel>)
}
}
}