change engine naming due to update to tensorrt 8.5.2

mafiosnik777 · Feb 2, 2023 · 9a07704 · 9a07704
1 parent e56804b
commit 9a07704
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/src/js/interpolation.js b/src/js/interpolation.js
@@ -159,7 +159,7 @@ class Interpolation {
             let onnx = getOnnx();
             // get engine path
             function getEnginePath() {
-                return path.join(appDataPath, '/.enhancr/models/engine', 'cain' + '_' + path.basename(onnx, '.onnx') + '_' + width + 'x' + height + '_' + fp + '.engine');
+                return path.join(appDataPath, '/.enhancr/models/engine', 'cain' + '_' + path.basename(onnx, '.onnx') + '_' + width + 'x' + height + '_' + fp + '_trt_8.5.2.engine');
             }
             let engineOut = getEnginePath();
             sessionStorage.setItem('engineOut', engineOut);
@@ -257,15 +257,15 @@ class Interpolation {
             function getRifeEngine() {
                 if (document.getElementById('rife-tta-check').checked) {
                     if (fp16.checked == true) {
-                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleTrue_fp16_${shapeDimensionsMax}.engine`);
+                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleTrue_fp16_${shapeDimensionsMax}` + '_trt_8.5.2.engine');
                     } else {
-                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleTrue_${shapeDimensionsMax}.engine`);
+                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleTrue_${shapeDimensionsMax}` + '_trt_8.5.2.engine');
                     }
                 } else {
                     if (fp16.checked == true) {
-                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleFalse_fp16_${shapeDimensionsMax}.engine`);
+                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleFalse_fp16_${shapeDimensionsMax}` + '_trt_8.5.2.engine');
                     } else {
-                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleFalse_${shapeDimensionsMax}.engine`);
+                        return path.join(appDataPath, '/.enhancr/models/engine', `rife46_ensembleFalse_${shapeDimensionsMax}` + '_trt_8.5.2.engine');
                     }
                 }
             }

diff --git a/src/js/restoration.js b/src/js/restoration.js
@@ -185,7 +185,7 @@ class Restoration {
                         return path.join(appDataPath, '/.enhancr/models/RealESRGAN', document.getElementById('custom-model-text').innerHTML);
                     }
                 } else {
-                    return path.join(appDataPath, '/.enhancr/models/engine', path.parse(onnx).name + '-' + fp + '_' + shapeDimensionsMax + '.engine');
+                    return path.join(appDataPath, '/.enhancr/models/engine', path.parse(onnx).name + '-' + fp + '_' + shapeDimensionsMax + '_trt_8.5.2.engine');
                 }
             }
             let engineOut = getEnginePath();

diff --git a/src/js/upscaling.js b/src/js/upscaling.js
@@ -188,7 +188,7 @@ class Upscaling {
                         return path.join(appDataPath, '/.enhancr/models/RealESRGAN', document.getElementById('custom-model-text').innerHTML);
                     }
                 } else {
-                    return path.join(appDataPath, '/.enhancr/models/engine', path.parse(onnx).name + '-' + fp + '_' + shapeDimensionsMax + '.engine');
+                    return path.join(appDataPath, '/.enhancr/models/engine', path.parse(onnx).name + '-' + fp + '_' + shapeDimensionsMax + '_trt_8.5.2.engine');
                 }
             }
             let engineOut = getEnginePath();