Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-13 13:47:36 +02:00
Code Issues Projects Releases Wiki Activity

semantic search: use configured device

Browse Source
This commit is contained in:
baudneo 2025-07-31 22:31:44 -06:00
parent d60d605e69
commit 58f509b379
No known key found for this signature in database
GPG Key ID: 51445F2ED08EBC7F
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/embeddings/embeddings.py
View File

@ -107,9 +107,7 @@ class Embeddings:
self.embedding = JinaV2Embedding( self.embedding = JinaV2Embedding(
model_size=self.config.semantic_search.model_size, model_size=self.config.semantic_search.model_size,
requestor=self.requestor, requestor=self.requestor,
device="GPU" device=self.config.semantic_search.device,
if self.config.semantic_search.model_size == "large"
else "CPU",
) )
self.text_embedding = lambda input_data: self.embedding( self.text_embedding = lambda input_data: self.embedding(
input_data, embedding_type="text" input_data, embedding_type="text"
@ -126,7 +124,7 @@ class Embeddings:
self.vision_embedding = JinaV1ImageEmbedding( self.vision_embedding = JinaV1ImageEmbedding(
model_size=config.semantic_search.model_size, model_size=config.semantic_search.model_size,
requestor=self.requestor, 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: def update_stats(self) -> None:
@ -416,4 +414,4 @@ class Embeddings:
finally: finally:
with self.reindex_lock: with self.reindex_lock:
self.reindex_running = False self.reindex_running = False
self.reindex_thread = None self.reindex_thread = None