From 54c2decce3468864dd9c69d96c95a480841955bc Mon Sep 17 00:00:00 2001 From: baudneo <86508179+baudneo@users.noreply.github.com> Date: Thu, 31 Jul 2025 22:36:54 -0600 Subject: [PATCH] remove self.model_size from ONNXModelRunner params --- frigate/embeddings/onnx/jina_v1_embedding.py | 2 -- frigate/embeddings/onnx/jina_v2_embedding.py | 1 - 2 files changed, 3 deletions(-) diff --git a/frigate/embeddings/onnx/jina_v1_embedding.py b/frigate/embeddings/onnx/jina_v1_embedding.py index b448ec816..d327fa8ba 100644 --- a/frigate/embeddings/onnx/jina_v1_embedding.py +++ b/frigate/embeddings/onnx/jina_v1_embedding.py @@ -128,7 +128,6 @@ class JinaV1TextEmbedding(BaseEmbedding): self.runner = ONNXModelRunner( os.path.join(self.download_path, self.model_file), self.device, - self.model_size, ) def _preprocess_inputs(self, raw_inputs): @@ -207,7 +206,6 @@ class JinaV1ImageEmbedding(BaseEmbedding): self.runner = ONNXModelRunner( os.path.join(self.download_path, self.model_file), self.device, - self.model_size, ) def _preprocess_inputs(self, raw_inputs): diff --git a/frigate/embeddings/onnx/jina_v2_embedding.py b/frigate/embeddings/onnx/jina_v2_embedding.py index e9def9a07..50b503d76 100644 --- a/frigate/embeddings/onnx/jina_v2_embedding.py +++ b/frigate/embeddings/onnx/jina_v2_embedding.py @@ -128,7 +128,6 @@ class JinaV2Embedding(BaseEmbedding): self.runner = ONNXModelRunner( os.path.join(self.download_path, self.model_file), self.device, - self.model_size, ) def _preprocess_image(self, image_data: bytes | Image.Image) -> np.ndarray: