Fixes for TensorRT

This commit is contained in:
cmdr2
2023-08-01 11:49:30 +05:30
parent 8538a684e7
commit d39e1da183
3 changed files with 35 additions and 3 deletions

View File

@@ -1346,9 +1346,19 @@ function createTask(task) {
function getCurrentUserRequest() {
const numOutputsTotal = parseInt(numOutputsTotalField.value)
const numOutputsParallel = parseInt(numOutputsParallelField.value)
let numOutputsParallel = parseInt(numOutputsParallelField.value)
const seed = randomSeedField.checked ? Math.floor(Math.random() * (2 ** 32 - 1)) : parseInt(seedField.value)
if (
testDiffusers.checked &&
document.getElementById("toggle-tensorrt-install").innerHTML == "Uninstall" &&
document.querySelector("#convert_to_tensorrt").checked
) {
// TRT enabled
numOutputsParallel = 1 // force 1 parallel
}
const newTask = {
batchesDone: 0,
numOutputsTotal: numOutputsTotal,