mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-26 19:06:11 +01:00
54eb03d2a1
* Add config option to select fp16 or quantized jina vision model * requires_fp16 for text and large models only * fix model type check * fix cpu * pass model size
19 lines
527 B
Python
19 lines
527 B
Python
from typing import Optional
|
|
|
|
from pydantic import Field
|
|
|
|
from .base import FrigateBaseModel
|
|
|
|
__all__ = ["SemanticSearchConfig"]
|
|
|
|
|
|
class SemanticSearchConfig(FrigateBaseModel):
|
|
enabled: bool = Field(default=False, title="Enable semantic search.")
|
|
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."
|
|
)
|