Add local model asset catalog
This commit is contained in:
+4
-3
@@ -145,7 +145,8 @@ GEOINTEL_INSTALL_AI=true
|
||||
|
||||
For Unraid/all-in-one deployments, place model files under
|
||||
`GEOINTEL_MODELS_PATH` so they appear in the container under `/app/models`, then
|
||||
set `YOLO_ENABLED=true` and `YOLO_MODEL_PATH=/app/models/<model>.pt`.
|
||||
set `YOLO_ENABLED=true`, `YOLO_MODELS_DIR=/app/models` and
|
||||
`YOLO_MODEL_PATH=/app/models/<model>.pt`.
|
||||
|
||||
Configure the Unraid/Tower env file from an existing local model without
|
||||
downloading weights or running inference:
|
||||
@@ -168,8 +169,8 @@ python scripts/configure_yolo_model.py \
|
||||
|
||||
The configurator refuses to proceed when no model exists or when multiple model
|
||||
files are present without `--model-file`. It writes only
|
||||
`GEOINTEL_INSTALL_AI=true`, `YOLO_ENABLED=true` and the mounted
|
||||
`YOLO_MODEL_PATH`.
|
||||
`GEOINTEL_INSTALL_AI=true`, `YOLO_ENABLED=true`, `YOLO_MODELS_DIR=/app/models`
|
||||
and the mounted `YOLO_MODEL_PATH`.
|
||||
|
||||
Clean old offline demo export artifacts without touching uploaded source data:
|
||||
|
||||
|
||||
@@ -9,7 +9,7 @@ from typing import Iterable
|
||||
|
||||
|
||||
SUPPORTED_MODEL_SUFFIXES = {".pt", ".onnx", ".engine"}
|
||||
ENV_KEYS = ("GEOINTEL_INSTALL_AI", "YOLO_ENABLED", "YOLO_MODEL_PATH")
|
||||
ENV_KEYS = ("GEOINTEL_INSTALL_AI", "YOLO_ENABLED", "YOLO_MODELS_DIR", "YOLO_MODEL_PATH")
|
||||
|
||||
|
||||
def _candidate_paths(models_dir: Path) -> list[Path]:
|
||||
@@ -149,6 +149,7 @@ def configure(args: argparse.Namespace) -> tuple[int, dict[str, object]]:
|
||||
updates = {
|
||||
"GEOINTEL_INSTALL_AI": "true",
|
||||
"YOLO_ENABLED": "true",
|
||||
"YOLO_MODELS_DIR": args.container_model_dir.rstrip("/"),
|
||||
"YOLO_MODEL_PATH": selected_container_path,
|
||||
}
|
||||
payload.update(
|
||||
|
||||
Reference in New Issue
Block a user