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 model type check

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

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

@ -56,8 +56,8 @@ class GenericONNXEmbedding:
self.providers, self.provider_options = get_ort_providers( self.providers, self.provider_options = get_ort_providers(
force_cpu=self.config.device == "CPU", force_cpu=self.config.device == "CPU",
requires_fp16=self.config.model_size == "large" requires_fp16=self.config.model_size == "large"
or self.config.model_type == "text", or self.model_type == "text",
openvino_device=self.config.device, openvino_device=device,
) )
self.download_path = os.path.join(MODEL_CACHE_DIR, self.model_name) self.download_path = os.path.join(MODEL_CACHE_DIR, self.model_name)