From 980a889001ee82d913b08d9ecd5dd3df7d508ab0 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Fri, 4 Oct 2024 16:06:04 -0500
Subject: [PATCH] install and add comment about transformers

---
 docker/main/requirements-wheels.txt          | 1 +
 frigate/embeddings/functions/minilm_l6_v2.py | 4 ++++
 2 files changed, 5 insertions(+)

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