diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt index f7717c808..3148848ca 100644 --- a/docker/main/requirements-wheels.txt +++ b/docker/main/requirements-wheels.txt @@ -33,6 +33,7 @@ unidecode == 1.3.* openvino == 2024.3.* # Embeddings sqlite_vec == 0.1.3 +transformers == 4.45.* onnx_clip == 4.0.* # Generative AI google-generativeai == 0.6.* diff --git a/frigate/embeddings/functions/minilm_l6_v2.py b/frigate/embeddings/functions/minilm_l6_v2.py index 651a64093..6a0e2d5ef 100644 --- a/frigate/embeddings/functions/minilm_l6_v2.py +++ b/frigate/embeddings/functions/minilm_l6_v2.py @@ -9,6 +9,10 @@ from typing import List import numpy as np import onnxruntime as ort import requests + +# importing this without pytorch or others causes a warning +# https://github.com/huggingface/transformers/issues/27214 +# suppressed by setting env TRANSFORMERS_NO_ADVISORY_WARNINGS=1 from transformers import AutoTokenizer from frigate.const import MODEL_CACHE_DIR