Beit: Add the gen_relative_position_index() function (#2306)

Co-authored-by: v-espitalier <>
This commit is contained in:
v-espitalier
2024-07-04 09:45:26 +02:00
committed by GitHub
parent 7f1ba8038c
commit ecff05d72b
2 changed files with 64 additions and 27 deletions

View File

@ -55,7 +55,7 @@ pub fn main() -> anyhow::Result<()> {
None => {
let api = hf_hub::api::sync::Api::new()?;
let api = api.model("vincent-espitalier/candle-beit".into());
api.get("beit_base_patch16_384.in22k_ft_in22k_in1k_adapted.safetensors")?
api.get("beit_base_patch16_384.in22k_ft_in22k_in1k.safetensors")?
}
Some(model) => model.into(),
};