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

install and add comment about transformers

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-04 16:06:04 -05:00
parent afb36a5ce9
commit 980a889001
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docker/main/requirements-wheels.txt
View File

@ -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.*

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

@ -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