chore(searcher): clean up deps

searcher/Cargo.toml

@@ -6,11 +6,9 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-candle-core = "0.1.0"
-candle-nn = "0.1.0"
-candle-examples = "0.1.0"
+candle-core = { git = "https://github.com/huggingface/candle.git" }
+candle-nn = { git = "https://github.com/huggingface/candle.git" }
 tokenizers = { version = "0.13.3", default-features = true }
-
 anyhow = { version = "1", features = ["backtrace"] }
 serde = { version = "1.0.171", features = ["derive"] }
 serde_json = "1.0.99"

searcher/src/args.rs

@@ -1,6 +1,7 @@
 use crate::embed;
 
 use anyhow::{anyhow, Error as E, Result};
+use candle_core::Device;
 use candle_nn::VarBuilder;
 use clap::Parser;
 use embed::model::{BertModel, Config, DTYPE};
@@ -24,7 +25,7 @@ pub struct Args {
 
 impl Args {
     pub fn build_model_and_tokenizer(&self) -> Result<(BertModel, Tokenizer)> {
-        let device = candle_examples::device(true)?;
+        let device = Device::Cpu;
         let default_model = "sentence-transformers/all-MiniLM-L6-v2".to_string();
 
         // source: https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/discussions/21