From c1bfc1df678a72804876a889461afce4d4ac4d27 Mon Sep 17 00:00:00 2001
From: Austin Kirsch <37373320+kirsch33@users.noreply.github.com>
Date: Sun, 10 Nov 2024 17:23:32 -0500
Subject: [PATCH] fix tensorrt model generation variable (#14902)

---
 .../detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/tensorrt/detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run b/docker/tensorrt/detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run
index 4d734e05a..a88da89d6 100755
--- a/docker/tensorrt/detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run
+++ b/docker/tensorrt/detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run
@@ -11,6 +11,7 @@ set -o errexit -o nounset -o pipefail
 MODEL_CACHE_DIR=${MODEL_CACHE_DIR:-"/config/model_cache/tensorrt"}
 TRT_VER=${TRT_VER:-$(cat /etc/TENSORRT_VER)}
 OUTPUT_FOLDER="${MODEL_CACHE_DIR}/${TRT_VER}"
+YOLO_MODELS=${YOLO_MODELS:-""}
 
 # Create output folder
 mkdir -p ${OUTPUT_FOLDER}