chore(代码变更): 1. 清理无用代码 2. 更新依赖库 3. 优化项目结构 4. 修复小型bug 5. 增加注释以提高可读性

This commit is contained in:
2026-07-08 10:44:32 +08:00
parent 4b342ef62f
commit d88afb5fe7
22 changed files with 1068 additions and 654372 deletions
@@ -0,0 +1,29 @@
pub mod q8;
use burn::module::Module;
use burn::record::{FullPrecisionSettings, NamedMpkFileRecorder};
use burn::tensor::backend::Backend;
use minicpm_core::config::LlamaConfig;
use minicpm_core::model::LlamaForCausalLM;
use self::q8::load_model_q8;
pub fn load_model<B: Backend>(
model_path: &str,
config: &LlamaConfig,
device: &B::Device,
) -> anyhow::Result<LlamaForCausalLM<B>> {
let model = LlamaForCausalLM::<B>::new(config.clone(), device);
let recorder = NamedMpkFileRecorder::<FullPrecisionSettings>::new();
let model = model.load_file(model_path, &recorder, device)?;
Ok(model)
}
pub fn load_model_q8_from_dir<B: Backend>(
model_dir: &str,
config: &LlamaConfig,
device: &B::Device,
) -> anyhow::Result<LlamaForCausalLM<B>> {
load_model_q8(model_dir, config, device)
}