tckm.yan/aha
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update fmt

Browse Source
This commit is contained in:
jhqxxx
2025-12-11 23:37:47 +08:00
parent 93bbed2aab
commit b3c6219e93
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/weight_test.rs
+3 -3
View File
@@ -51,12 +51,12 @@ fn voxcpm1_5_weight() -> Result<()> {
let model_path = "/home/jhq/huggingface_model/OpenBMB/VoxCPM1.5/"; let model_path = "/home/jhq/huggingface_model/OpenBMB/VoxCPM1.5/";
let model_list = find_type_files(model_path, "pth")?; let model_list = find_type_files(model_path, "pth")?;
println!("model_list: {:?}", model_list); println!("model_list: {:?}", model_list);
let dev = get_device(None); // let dev = get_device(None);
let mut dict_to_hashmap = HashMap::new(); 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 { for m in model_list {
let dict = read_all_with_key(m, Some("state_dict"))?; 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 { for (k, v) in dict {
println!("key: {}, tensor shape: {:?}", k, v); println!("key: {}, tensor shape: {:?}", k, v);
dict_to_hashmap.insert(k, v); dict_to_hashmap.insert(k, v);