Fix the helium weights download. (#2717)
This commit is contained in:
parent
158817f230
commit
efd0e6822f
|
@ -248,7 +248,7 @@ fn main() -> Result<()> {
|
|||
.split(',')
|
||||
.map(std::path::PathBuf::from)
|
||||
.collect::<Vec<_>>(),
|
||||
None => candle_examples::hub_load_safetensors(&repo, "model.safetensors")?,
|
||||
None => vec![repo.get("model.safetensors")?],
|
||||
};
|
||||
println!("retrieved the files in {:?}", start.elapsed());
|
||||
let tokenizer = Tokenizer::from_file(tokenizer_filename).map_err(E::msg)?;
|
||||
|
|
Loading…
Reference in New Issue