add paddleocr-vl

This commit is contained in:
jhqxxx
2025-12-09 00:41:30 +08:00
parent aef4d7d34d
commit 4228616058
19 changed files with 1498 additions and 83 deletions
+12 -2
View File
@@ -1,7 +1,8 @@
use aha::models::{
deepseek_ocr::config::DeepseekOCRConfig, hunyuan_ocr::config::HunYuanVLConfig,
minicpm4::config::MiniCPM4Config, qwen2_5vl::config::Qwen2_5VLConfig,
qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig,
minicpm4::config::MiniCPM4Config, paddleocr_vl::config::PaddleOCRVLConfig,
qwen2_5vl::config::Qwen2_5VLConfig, qwen3vl::config::Qwen3VLConfig,
voxcpm::config::VoxCPMConfig,
};
use anyhow::Result;
@@ -65,3 +66,12 @@ fn hunyuan_ocr_config() -> Result<()> {
println!("{:?}", config);
Ok(())
}
#[test]
fn paddleocr_vl_config() -> Result<()> {
// cargo test -F cuda paddleocr_vl_config -r -- --nocapture
let model_path = "/home/jhq/huggingface_model/PaddlePaddle/PaddleOCR-VL/";
let config_path = model_path.to_string() + "/config.json";
let config: PaddleOCRVLConfig = serde_json::from_slice(&std::fs::read(config_path)?)?;
println!("{:?}", config);
Ok(())
}