add PaddleOCR-VL-1.6
This commit is contained in:
+3
-1
@@ -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 => {
|
||||
|
||||
@@ -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
@@ -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))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user