extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-28 01:48:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow batch size ranges again in TRT

Browse Source
This commit is contained in:
cmdr2 2023-08-02 14:03:59 +05:30
parent 35073adc1f
commit d1fdf1766a
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/media/js/main.js
View File

@ -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: [],
}

3
ui/media/js/parameters.js
View File

@ -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/>