diff options
author | Lia Lenckowski <lialenck@protonmail.com> | 2024-11-27 15:59:54 +0100 |
---|---|---|
committer | Lia Lenckowski <lialenck@protonmail.com> | 2024-11-27 15:59:54 +0100 |
commit | 4a278bb7fb9ba37240d2778132400a5e97b79917 (patch) | |
tree | f88619823c9dfc720c99dfa64a856d958c15e4a0 /src/embedders/ai.rs | |
parent | 58c1da21ec0e7a2a52cdc84469fe05d25bd94e13 (diff) | |
download | embeddings-sort-4a278bb7fb9ba37240d2778132400a5e97b79917.tar embeddings-sort-4a278bb7fb9ba37240d2778132400a5e97b79917.tar.bz2 embeddings-sort-4a278bb7fb9ba37240d2778132400a5e97b79917.tar.zst |
windows compat: replace xdg crate with dirs, and conditionally compile symlink output
Diffstat (limited to 'src/embedders/ai.rs')
-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..."); |