From b3c6219e93e244d8f07a07066a798b4385a46631 Mon Sep 17 00:00:00 2001 From: jhqxxx Date: Thu, 11 Dec 2025 23:37:47 +0800 Subject: [PATCH] update fmt --- tests/weight_test.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/weight_test.rs b/tests/weight_test.rs index dc98c68..ab68db6 100644 --- a/tests/weight_test.rs +++ b/tests/weight_test.rs @@ -51,12 +51,12 @@ fn voxcpm1_5_weight() -> Result<()> { let model_path = "/home/jhq/huggingface_model/OpenBMB/VoxCPM1.5/"; let model_list = find_type_files(model_path, "pth")?; println!("model_list: {:?}", model_list); - let dev = get_device(None); + // let dev = get_device(None); let mut dict_to_hashmap = HashMap::new(); - let mut dtype = candle_core::DType::F32; + // let mut dtype = candle_core::DType::F32; for m in model_list { let dict = read_all_with_key(m, Some("state_dict"))?; - dtype = dict[0].1.dtype(); + // dtype = dict[0].1.dtype(); for (k, v) in dict { println!("key: {}, tensor shape: {:?}", k, v); dict_to_hashmap.insert(k, v);