From 58f509b3799f267ccff778f06f4682757f19da93 Mon Sep 17 00:00:00 2001
From: baudneo <86508179+baudneo@users.noreply.github.com>
Date: Thu, 31 Jul 2025 22:31:44 -0600
Subject: [PATCH] semantic search: use configured device

---
 frigate/embeddings/embeddings.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py
index 833ab9ab2..142545733 100644
--- a/frigate/embeddings/embeddings.py
+++ b/frigate/embeddings/embeddings.py
@@ -107,9 +107,7 @@ class Embeddings:
             self.embedding = JinaV2Embedding(
                 model_size=self.config.semantic_search.model_size,
                 requestor=self.requestor,
-                device="GPU"
-                if self.config.semantic_search.model_size == "large"
-                else "CPU",
+                device=self.config.semantic_search.device,
             )
             self.text_embedding = lambda input_data: self.embedding(
                 input_data, embedding_type="text"
@@ -126,7 +124,7 @@ class Embeddings:
             self.vision_embedding = JinaV1ImageEmbedding(
                 model_size=config.semantic_search.model_size,
                 requestor=self.requestor,
-                device="GPU" if config.semantic_search.model_size == "large" else "CPU",
+                device=self.config.semantic_search.device,
             )
 
     def update_stats(self) -> None:
@@ -416,4 +414,4 @@ class Embeddings:
         finally:
             with self.reindex_lock:
                 self.reindex_running = False
-                self.reindex_thread = None
+                self.reindex_thread = None
\ No newline at end of file