administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix cpu

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-10 17:33:36 -05:00
parent 49fddeaeb3
commit cdb794ece8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/embeddings/functions/onnx.py
View File

@ -54,7 +54,7 @@ class GenericONNXEmbedding:
self.embedding_function = embedding_function self.embedding_function = embedding_function
self.model_type = model_type # 'text' or 'vision' self.model_type = model_type # 'text' or 'vision'
self.providers, self.provider_options = get_ort_providers( self.providers, self.provider_options = get_ort_providers(
force_cpu=self.config.device == "CPU", force_cpu=device == "CPU",
requires_fp16=self.config.model_size == "large" requires_fp16=self.config.model_size == "large"
or self.model_type == "text", or self.model_type == "text",
openvino_device=device, openvino_device=device,