diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2025-01-13 18:21:37 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-01-13 18:21:37 +0100 |
| commit | efd0e6822f4d0e2433f0ae02ba16f16cda834d97 (patch) | |
| tree | c45e1e02d6699216c4c21a66fc1c0355bde0f452 | |
| parent | 158817f230095f4a3599a29c30c0a3ae48c10b01 (diff) | |
| download | candle-efd0e6822f4d0e2433f0ae02ba16f16cda834d97.tar.gz candle-efd0e6822f4d0e2433f0ae02ba16f16cda834d97.tar.bz2 candle-efd0e6822f4d0e2433f0ae02ba16f16cda834d97.zip | |
Fix the helium weights download. (#2717)
| -rw-r--r-- | candle-examples/examples/helium/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/helium/main.rs b/candle-examples/examples/helium/main.rs index 8cf63758..31f949bf 100644 --- a/candle-examples/examples/helium/main.rs +++ b/candle-examples/examples/helium/main.rs @@ -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)?; |