chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

Remove device config and use model size to configure device used (#14290)

Browse Source 
* Remove device config and use model size to configure device used

* Don't show Frigate+ submission when in progress

* Add docs link for bounding box colors
This commit is contained in:
Nicolas Mowen
2024-10-11 16:08:14 -06:00
committed by GitHub
parent 8a8a0c7dec
commit 6e332bbdf8
6 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
frigate/config/semantic_search.py
View File

@@ -12,7 +12,6 @@ class SemanticSearchConfig(FrigateBaseModel):
reindex: Optional[bool] = Field(
default=False, title="Reindex all detections on startup."
)
device: str = Field(default="AUTO", title="Device Type")
model_size: str = Field(
default="small", title="The size of the embeddings model used."
)

2
frigate/embeddings/embeddings.py
View File

@@ -127,7 +127,7 @@ class Embeddings:
model_size=config.model_size,
model_type="vision",
requestor=self.requestor,
device=self.config.device,
device="GPU" if config.model_size == "large" else "CPU",
)
def upsert_thumbnail(self, event_id: str, thumbnail: bytes):