diff --git a/frigate/embeddings/onnx/jina_v2_embedding_ax.py b/frigate/embeddings/onnx/jina_v2_embedding_ax.py
index 1d39ce014..be222c306 100644
--- a/frigate/embeddings/onnx/jina_v2_embedding_ax.py
+++ b/frigate/embeddings/onnx/jina_v2_embedding_ax.py
@@ -4,21 +4,18 @@ import io
 import logging
 import os
 import threading
-from typing import Any
 
+import axengine as axe
 import numpy as np
 from PIL import Image
 from transformers import AutoTokenizer
 from transformers.utils.logging import disable_progress_bar, set_verbosity_error
 
-from frigate.const import MODEL_CACHE_DIR
-from frigate.embeddings.onnx.base_embedding import BaseEmbedding
 from frigate.comms.inter_process import InterProcessRequestor
-from frigate.util.downloader import ModelDownloader
-from frigate.types import ModelStatusTypesEnum
 from frigate.const import MODEL_CACHE_DIR, UPDATE_MODEL_STATE
-
-import axengine as axe
+from frigate.embeddings.onnx.base_embedding import BaseEmbedding
+from frigate.types import ModelStatusTypesEnum
+from frigate.util.downloader import ModelDownloader
 
 # disables the progress bar and download logging for downloading tokenizers and image processors
 disable_progress_bar()