administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-07 22:15:28 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

use an enum for model_size

Browse Source 
the frontend will then render this as a select dropdown because of the changes in the json schema
This commit is contained in:
Josh Hawkins 2026-05-06 13:08:38 -05:00
parent 25c24ab5e8
commit f02f7290f0
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
frigate/config/classification.py
View File

@ -26,6 +26,11 @@ class EnrichmentsDeviceEnum(str, Enum):
CPU = "CPU" CPU = "CPU"
class ModelSizeEnum(str, Enum):
small = "small"
large = "large"
class TriggerType(str, Enum): class TriggerType(str, Enum):
THUMBNAIL = "thumbnail" THUMBNAIL = "thumbnail"
DESCRIPTION = "description" DESCRIPTION = "description"
@ -53,13 +58,13 @@ class AudioTranscriptionConfig(FrigateBaseModel):
title="Transcription language", title="Transcription language",
description="Language code used for transcription/translation (for example 'en' for English). See https://whisper-api.com/docs/languages/ for supported language codes.", description="Language code used for transcription/translation (for example 'en' for English). See https://whisper-api.com/docs/languages/ for supported language codes.",
) )
device: Optional[EnrichmentsDeviceEnum] = Field( device: EnrichmentsDeviceEnum = Field(
default=EnrichmentsDeviceEnum.CPU, default=EnrichmentsDeviceEnum.CPU,
title="Transcription device", title="Transcription device",
description="Device key (CPU/GPU) to run the transcription model on. Only NVIDIA CUDA GPUs are currently supported for transcription.", description="Device key (CPU/GPU) to run the transcription model on. Only NVIDIA CUDA GPUs are currently supported for transcription.",
) )
model_size: str = Field( model_size: ModelSizeEnum = Field(
default="small", default=ModelSizeEnum.small,
title="Model size", title="Model size",
description="Model size to use for offline audio event transcription.", description="Model size to use for offline audio event transcription.",
) )
@ -189,8 +194,8 @@ class SemanticSearchConfig(FrigateBaseModel):
return v return v
return v return v
model_size: str = Field( model_size: ModelSizeEnum = Field(
default="small", default=ModelSizeEnum.small,
title="Model size", title="Model size",
description="Select model size; 'small' runs on CPU and 'large' typically requires GPU.", description="Select model size; 'small' runs on CPU and 'large' typically requires GPU.",
) )
@ -253,8 +258,8 @@ class FaceRecognitionConfig(FrigateBaseModel):
title="Enable face recognition", title="Enable face recognition",
description="Enable or disable face recognition for all cameras; can be overridden per-camera.", description="Enable or disable face recognition for all cameras; can be overridden per-camera.",
) )
model_size: str = Field( model_size: ModelSizeEnum = Field(
default="small", default=ModelSizeEnum.small,
title="Model size", title="Model size",
description="Model size to use for face embeddings (small/large); larger may require GPU.", description="Model size to use for face embeddings (small/large); larger may require GPU.",
) )
@ -335,8 +340,8 @@ class LicensePlateRecognitionConfig(FrigateBaseModel):
title="Enable LPR", title="Enable LPR",
description="Enable or disable license plate recognition for all cameras; can be overridden per-camera.", description="Enable or disable license plate recognition for all cameras; can be overridden per-camera.",
) )
model_size: str = Field( model_size: ModelSizeEnum = Field(
default="small", default=ModelSizeEnum.small,
title="Model size", title="Model size",
description="Model size used for text detection/recognition. Most users should use 'small'.", description="Model size used for text detection/recognition. Most users should use 'small'.",
) )