add glm-asr-nano
This commit is contained in:
@@ -119,3 +119,22 @@ fn hunyuanocr_weight() -> Result<()> {
|
||||
println!("model_list: {:?}", model_list);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn glm_asr_nano_weight() -> Result<()> {
|
||||
let model_path = "/home/jhq/huggingface_model/ZhipuAI/GLM-ASR-Nano-2512/";
|
||||
let model_list = find_type_files(model_path, "safetensors")?;
|
||||
|
||||
let device = Device::Cpu;
|
||||
for m in &model_list {
|
||||
let weights = safetensors::load(m, &device)?;
|
||||
for (key, tensor) in weights.iter() {
|
||||
if key.contains(".embed_tokens") {
|
||||
println!("=== {} === {:?}", key, tensor.shape());
|
||||
}
|
||||
// println!("=== {} === {:?}", key, tensor.shape());
|
||||
}
|
||||
}
|
||||
println!("model_list: {:?}", model_list);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user