diff options
Diffstat (limited to 'src/embedders')
-rw-r--r-- | src/embedders/ai.rs | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/embedders/ai.rs b/src/embedders/ai.rs index 8c9de11..7d31a6b 100644 --- a/src/embedders/ai.rs +++ b/src/embedders/ai.rs @@ -13,7 +13,10 @@ pub(crate) struct ContentEmbedder<'a, Metric> { } impl<'a, Metric> ContentEmbedder<'a, Metric> { pub(crate) fn new(cfg: &'a Config) -> Self { - ContentEmbedder { cfg, _sim: PhantomData } + ContentEmbedder { + cfg, + _sim: PhantomData, + } } } @@ -23,11 +26,13 @@ impl<Metric: VecMetric> ContentEmbedder<'_, Metric> { paths: &[PathBuf], ) -> Result<Vec<Result<<Self as BatchEmbedder>::Embedding>>> { let mut options = ImageInitOptions::default(); - options.cache_dir = self.cfg.base_dirs.get_cache_home(); + options.cache_dir = self.cfg.cache_dir.join("models"); let embedder = ImageEmbedding::try_new(options)?; let bar = ProgressBar::new(paths.len() as u64); - bar.set_style(ProgressStyle::with_template("{bar:20.cyan/blue} {pos}/{len} {msg}")?); + bar.set_style(ProgressStyle::with_template( + "{bar:20.cyan/blue} {pos}/{len} {msg}", + )?); bar.enable_steady_tick(std::time::Duration::from_millis(100)); bar.set_message("Embedding images..."); |