From d1fdf1766a81bdbf20f159032372d635a39b426b Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 2 Aug 2023 14:03:59 +0530
Subject: [PATCH] Allow batch size ranges again in TRT

---
 ui/media/js/main.js       | 4 ++--
 ui/media/js/parameters.js | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 3ec2723c..21e55923 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -1454,8 +1454,8 @@ function getCurrentUserRequest() {
         newTask.reqBody.convert_to_tensorrt = document.querySelector("#convert_to_tensorrt").checked
         let trtBuildConfig = {
             batch_size_range: [
-                parseInt(document.querySelector("#trt-build-batch").value),
-                parseInt(document.querySelector("#trt-build-batch").value),
+                parseInt(document.querySelector("#trt-build-min-batch").value),
+                parseInt(document.querySelector("#trt-build-max-batch").value),
             ],
             dimensions_range: [],
         }
diff --git a/ui/media/js/parameters.js b/ui/media/js/parameters.js
index 96bcad87..8b66c461 100644
--- a/ui/media/js/parameters.js
+++ b/ui/media/js/parameters.js
@@ -251,7 +251,8 @@ var PARAMETERS = [
                <b>Early access version:</b> support for LoRA is still under development.
                <div id="trt-build-config" class="displayNone">
                     <h3>Build Config:</h3>
-                    Batch size: <input id="trt-build-batch" type="number" min="1" value="1" /><br/><br/>
+                    Min batch size: <input id="trt-build-min-batch" type="number" min="1" value="1" /><br/>
+                    Max batch size: <input id="trt-build-max-batch" type="number" min="1" value="1" /><br/><br/>
                     <b>Build for resolutions</b>:<br/>
                     <input id="trt-build-res-512" type="checkbox" value="1" /> 512x512 to 768x768<br/>
                     <input id="trt-build-res-768" type="checkbox" value="1" checked /> 768x768 to 1024x1024<br/>