From efd0e6822f4d0e2433f0ae02ba16f16cda834d97 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Mon, 13 Jan 2025 18:21:37 +0100
Subject: [PATCH] Fix the helium weights download. (#2717)

---
 candle-examples/examples/helium/main.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-examples/examples/helium/main.rs b/candle-examples/examples/helium/main.rs
index 8cf63758c..31f949bf3 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)?;