Fix the helium weights download.

huggingface · Jan 13, 2025 · 8f162fc · 8f162fc
1 parent 158817f
commit 8f162fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)?;