Fix the helium weights download. (#2717)

This commit is contained in:
Laurent Mazare
2025-01-13 18:21:37 +01:00
committed by GitHub
parent 158817f230
commit efd0e6822f

View File

@ -248,7 +248,7 @@ fn main() -> Result<()> {
.split(',') .split(',')
.map(std::path::PathBuf::from) .map(std::path::PathBuf::from)
.collect::<Vec<_>>(), .collect::<Vec<_>>(),
None => candle_examples::hub_load_safetensors(&repo, "model.safetensors")?, None => vec![repo.get("model.safetensors")?],
}; };
println!("retrieved the files in {:?}", start.elapsed()); println!("retrieved the files in {:?}", start.elapsed());
let tokenizer = Tokenizer::from_file(tokenizer_filename).map_err(E::msg)?; let tokenizer = Tokenizer::from_file(tokenizer_filename).map_err(E::msg)?;