mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-08-04 01:48:54 +03:00
backend tweaks
This commit is contained in:
@@ -640,7 +640,7 @@ class FrigateConfig(FrigateBaseModel):
|
|||||||
# set notifications state
|
# set notifications state
|
||||||
self.notifications.enabled_in_config = self.notifications.enabled
|
self.notifications.enabled_in_config = self.notifications.enabled
|
||||||
|
|
||||||
# validate genai: each role (tools, vision, embeddings) at most once
|
# validate genai: each role (chat, descriptions, embeddings) at most once
|
||||||
role_to_name: dict[GenAIRoleEnum, str] = {}
|
role_to_name: dict[GenAIRoleEnum, str] = {}
|
||||||
for name, genai_cfg in self.genai.items():
|
for name, genai_cfg in self.genai.items():
|
||||||
for role in genai_cfg.roles:
|
for role in genai_cfg.roles:
|
||||||
|
|||||||
@@ -93,7 +93,7 @@ class ModelConfig(BaseModel):
|
|||||||
model_type: ModelTypeEnum = Field(
|
model_type: ModelTypeEnum = Field(
|
||||||
default=ModelTypeEnum.ssd,
|
default=ModelTypeEnum.ssd,
|
||||||
title="Object Detection Model Type",
|
title="Object Detection Model Type",
|
||||||
description="Detector model architecture type (ssd, yolox, yolonas) used by some detectors for optimization.",
|
description="Detector model architecture type (ssd, yolox, yolonas, yolo-generic, rfdetr, dfine) used by some detectors for optimization.",
|
||||||
)
|
)
|
||||||
_merged_labelmap: dict[int, str] | None = PrivateAttr()
|
_merged_labelmap: dict[int, str] | None = PrivateAttr()
|
||||||
_colormap: dict[int, tuple[int, int, int]] = PrivateAttr()
|
_colormap: dict[int, tuple[int, int, int]] = PrivateAttr()
|
||||||
|
|||||||
@@ -326,7 +326,7 @@ def get_ort_providers(
|
|||||||
{
|
{
|
||||||
"device_id": device_id,
|
"device_id": device_id,
|
||||||
"trt_fp16_enable": requires_fp16
|
"trt_fp16_enable": requires_fp16
|
||||||
and os.environ.get("USE_FP_16", "True") != "False",
|
and os.environ.get("USE_FP16", "True") != "False",
|
||||||
"trt_timing_cache_enable": True,
|
"trt_timing_cache_enable": True,
|
||||||
"trt_engine_cache_enable": True,
|
"trt_engine_cache_enable": True,
|
||||||
"trt_timing_cache_path": os.path.join(
|
"trt_timing_cache_path": os.path.join(
|
||||||
|
|||||||
Reference in New Issue
Block a user