use burn::backend::Wgpu;
use minicpm_convert::export_model;
use std::path::Path;

fn main() -> anyhow::Result<()> {
    let device = Default::default();

    let safetensors_path = Path::new("MiniCPM5-1B/model-00000-of-00001.safetensors");
    let config_path = Path::new("MiniCPM5-1B/config.json");
    let tokenizer_path = Path::new("MiniCPM5-1B/tokenizer.json");
    let output_dir = Path::new("model");

    export_model::<Wgpu>(
        safetensors_path,
        config_path,
        tokenizer_path,
        output_dir,
        &device,
    )?;

    println!("模型转换完成！");

    Ok(())
}