administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-27 17:17:40 +03:00
Code Issues Actions 2 Packages 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 cb03489a97
commit b8db85ec94
No known key found for this signature in database
GPG Key ID: 51445F2ED08EBC7F
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/embeddings/embeddings.py
View File

@ -112,9 +112,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"
@ -131,7 +129,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: