update code format
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
use aha::models::{
|
||||
minicpm4::config::MiniCPM4Config, qwen2_5vl::config::Qwen2_5VLConfig, qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig
|
||||
minicpm4::config::MiniCPM4Config, qwen2_5vl::config::Qwen2_5VLConfig,
|
||||
qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig,
|
||||
};
|
||||
use anyhow::Result;
|
||||
|
||||
|
||||
+1
-1
@@ -1,4 +1,4 @@
|
||||
use aha::utils::{tensor_utils::bitor_tensor};
|
||||
use aha::utils::tensor_utils::bitor_tensor;
|
||||
use anyhow::Result;
|
||||
use candle_core::Tensor;
|
||||
|
||||
|
||||
@@ -50,7 +50,7 @@ fn voxcpm_weight() -> Result<()> {
|
||||
fn qwen3vl_weight() -> Result<()> {
|
||||
let model_path = "/home/jhq/huggingface_model/Qwen/Qwen3-VL-4B-Instruct/";
|
||||
let model_list = find_type_files(model_path, "safetensors")?;
|
||||
|
||||
|
||||
let device = Device::Cpu;
|
||||
for m in &model_list {
|
||||
let weights = safetensors::load(m, &device)?;
|
||||
@@ -60,4 +60,4 @@ fn qwen3vl_weight() -> Result<()> {
|
||||
}
|
||||
println!("model_list: {:?}", model_list);
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user