diff --git a/docker/main/requirements-wheels-nodeps.txt b/docker/main/requirements-wheels-nodeps.txt index 62bca937b..bead6ca66 100644 --- a/docker/main/requirements-wheels-nodeps.txt +++ b/docker/main/requirements-wheels-nodeps.txt @@ -1,3 +1,32 @@ # Embeddings chromadb == 0.5.0 onnx_clip == 4.0.* + +# chromadb dependencies +bcrypt>=4.0.1 +chroma-hnswlib==0.7.6 +fastapi>=0.95.2 +graphlib_backport==1.0.3; python_version < '3.9' +grpcio>=1.58.0 +httpx>=0.27.0 +importlib-resources +kubernetes>=28.1.0 +mmh3>=4.0.1 +numpy>=1.22.5, <2.0.0 +opentelemetry-api>=1.2.0 +opentelemetry-exporter-otlp-proto-grpc>=1.24.0 +opentelemetry-instrumentation-fastapi>=0.41b0 +opentelemetry-sdk>=1.2.0 +orjson>=3.9.12, < 3.10.6 # 3.10.7 is currently missing a wheel for x86 glibc +overrides>=7.3.1 +posthog>=2.4.0 +pydantic>=1.9 +pypika>=0.48.9 +PyYAML>=6.0.0 +rich>=10.11.0 +tenacity>=8.2.3 +tokenizers>=0.13.2 +tqdm>=4.65.0 +typer>=0.9.0 +typing_extensions>=4.5.0 +uvicorn[standard]>=0.18.3 diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py index 998738d50..8179de15e 100644 --- a/frigate/embeddings/embeddings.py +++ b/frigate/embeddings/embeddings.py @@ -85,7 +85,10 @@ class Embeddings: @property def description(self) -> Collection: return self.client.get_or_create_collection( - name="event_description", embedding_function=MiniLMEmbedding() + name="event_description", + embedding_function=MiniLMEmbedding( + preferred_providers=["CPUExecutionProvider"] + ), ) def reindex(self) -> None: diff --git a/frigate/embeddings/functions/clip.py b/frigate/embeddings/functions/clip.py index 16d020cf0..6e44033df 100644 --- a/frigate/embeddings/functions/clip.py +++ b/frigate/embeddings/functions/clip.py @@ -69,7 +69,9 @@ class Clip(OnnxClip): try: if os.path.exists(path): - return ort.InferenceSession(path, providers=providers, options=options) + return ort.InferenceSession( + path, providers=providers, provider_options=options + ) else: raise FileNotFoundError( errno.ENOENT, @@ -101,7 +103,7 @@ class Clip(OnnxClip): # Finally move the temporary file to the correct location temporary_filename.rename(path) return ort.InferenceSession( - path, providers=provider, provider_options=optionsm + path, providers=provider, provider_options=options )