aboutsummaryrefslogtreecommitdiff
path: root/src/embedders/ai.rs
diff options
context:
space:
mode:
authorLia Lenckowski <lialenck@protonmail.com>2024-11-27 15:59:54 +0100
committerLia Lenckowski <lialenck@protonmail.com>2024-11-27 15:59:54 +0100
commit4a278bb7fb9ba37240d2778132400a5e97b79917 (patch)
treef88619823c9dfc720c99dfa64a856d958c15e4a0 /src/embedders/ai.rs
parent58c1da21ec0e7a2a52cdc84469fe05d25bd94e13 (diff)
downloadembeddings-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.rs11
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...");