holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

TRT ui changes

Browse Source
This commit is contained in:
cmdr2 2023-08-02 00:08:43 +05:30
parent fa58996f37
commit 38ab5b090f
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/check_modules.py
View File

@ -18,7 +18,7 @@ os_name = platform.system()
modules_to_check = {
"torch": ("1.11.0", "1.13.1", "2.0.0"),
"torchvision": ("0.12.0", "0.14.1", "0.15.1"),
"sdkit": "1.0.157",
"sdkit": "1.0.158",
"stable-diffusion-sdkit": "2.1.4",
"rich": "12.6.0",
"uvicorn": "0.19.0",

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-min-batch").value),
parseInt(document.querySelector("#trt-build-max-batch").value),
parseInt(document.querySelector("#trt-build-batch").value),
parseInt(document.querySelector("#trt-build-batch").value),
],
dimensions_range: [],
}

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

@ -251,8 +251,7 @@ 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>
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/>
Batch size: <input id="trt-build-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/>