loading a model

cn1.rs

let repo = Repo::with_revision(model_name.parse()?, RepoType::Model, revision.parse()?);
let api = Api::new()?;
let api = api.repo(repo);
let config_filename = api.get("config.json")?;
let tokenizer_filename = api.get("tokenizer.json")?;
let weights_filename = api.get("model.safetensors")?;
// load the model config
let config = std::fs::read_to_string(config_filename)?;
let config: Config = serde_json::from_str(&config)?;
// load the tokenizer
let tokenizer = Tokenizer::from_file(tokenizer_filename)
.map_err(anyhow::Error::msg)?;
// load the model
let vb =
    unsafe { VarBuilder::from_mmaped_safetensors(&[weights_filename], DTYPE, &device)? };
let model = BertModel::load(vb, &config)?;