add PaddleOCR-VL-1.6

This commit is contained in:
jhqxxx
2026-06-06 18:41:29 +08:00
parent cb2fe57359
commit a2d0f64c5b
11 changed files with 74 additions and 12 deletions
+3 -1
View File
@@ -298,7 +298,9 @@ pub(crate) fn run_run(args: RunArgs) -> anyhow::Result<()> {
WhichModel::HunyuanOCR => {
hunyuan_ocr::HunyuanORExec::run(&input, output.as_deref(), &weight_path)?;
}
WhichModel::PaddleOCRVL | WhichModel::PaddleOCRVL1_5 => {
WhichModel::PaddleOCRVL
| WhichModel::PaddleOCRVL1_5
| WhichModel::PaddleOCRVL1_6 => {
paddleocr_vl::PaddleOVLExec::run(&input, output.as_deref(), &weight_path)?;
}
WhichModel::RMBG2_0 => {
+4 -1
View File
@@ -72,6 +72,8 @@ pub enum WhichModel {
PaddleOCRVL,
#[value(name = "PaddlePaddle/PaddleOCR-VL-1.5")]
PaddleOCRVL1_5,
#[value(name = "PaddlePaddle/PaddleOCR-VL-1.6")]
PaddleOCRVL1_6,
#[value(name = "AI-ModelScope/RMBG-2.0")]
RMBG2_0,
#[value(name = "OpenBMB/VoxCPM-0.5B")]
@@ -165,7 +167,8 @@ impl WhichModel {
| WhichModel::HunyuanOCR
| WhichModel::GlmOCR
| WhichModel::PaddleOCRVL
| WhichModel::PaddleOCRVL1_5 => "ocr",
| WhichModel::PaddleOCRVL1_5
| WhichModel::PaddleOCRVL1_6 => "ocr",
// ASR models
WhichModel::Qwen3ASR0_6B
| WhichModel::Qwen3ASR1_7B
+3 -1
View File
@@ -284,7 +284,9 @@ pub fn load_model<'a>(
let model = HunyuanOCRGenerateModel::init(path, device, dtype)?;
ModelInstance::HunyuanOCR(model)
}
WhichModel::PaddleOCRVL | WhichModel::PaddleOCRVL1_5 => {
WhichModel::PaddleOCRVL
| WhichModel::PaddleOCRVL1_5
| WhichModel::PaddleOCRVL1_6 => {
let model = PaddleOCRVLGenerateModel::init(path, device, dtype)?;
ModelInstance::PaddleOCRVL(Box::new(model))
}