update fmt
This commit is contained in:
@@ -1,5 +1,8 @@
|
||||
use aha::models::{
|
||||
deepseek_ocr::config::DeepseekOCRConfig, hunyuan_ocr::config::HunYuanVLConfig, lfm2::config::Lfm2Config, minicpm4::config::MiniCPM4Config, paddleocr_vl::config::PaddleOCRVLConfig, qwen2_5vl::config::Qwen2_5VLConfig, qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig
|
||||
deepseek_ocr::config::DeepseekOCRConfig, hunyuan_ocr::config::HunYuanVLConfig,
|
||||
lfm2::config::Lfm2Config, minicpm4::config::MiniCPM4Config,
|
||||
paddleocr_vl::config::PaddleOCRVLConfig, qwen2_5vl::config::Qwen2_5VLConfig,
|
||||
qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig,
|
||||
};
|
||||
use anyhow::Result;
|
||||
|
||||
|
||||
+1
-1
@@ -43,7 +43,7 @@ fn messy_test() -> Result<()> {
|
||||
// aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
|
||||
// let model_path = format!("{}/deepseek-ai/DeepSeek-OCR-2/", save_dir);
|
||||
// let stem = std::path::Path::new(&model_path)
|
||||
// .file_name()
|
||||
// .file_name()
|
||||
// .and_then(|s| s.to_str())
|
||||
// .unwrap_or("qwen3.5");
|
||||
// println!("stem: {:?}", stem);
|
||||
|
||||
+6
-4
@@ -1,7 +1,10 @@
|
||||
use std::{pin::pin, time::Instant};
|
||||
use aha::{
|
||||
chat::ChatCompletionParameters,
|
||||
models::{GenerateModel, lfm2::generate::Lfm2GenerateModel},
|
||||
};
|
||||
use anyhow::Result;
|
||||
use aha::{chat::ChatCompletionParameters, models::{GenerateModel, lfm2::generate::Lfm2GenerateModel}};
|
||||
use rocket::futures::StreamExt;
|
||||
use std::{pin::pin, time::Instant};
|
||||
|
||||
#[test]
|
||||
fn lfm2_generate() -> Result<()> {
|
||||
@@ -44,7 +47,6 @@ fn lfm2_generate() -> Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
|
||||
#[tokio::test]
|
||||
async fn lfm2_stream() -> Result<()> {
|
||||
// test with cuda: RUST_BACKTRACE=1 cargo test -F cuda --test test_lfm2 lfm2_stream -r -- --nocapture
|
||||
@@ -88,4 +90,4 @@ async fn lfm2_stream() -> Result<()> {
|
||||
println!("Time elapsed in generate is: {:?}", i_duration);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user