fix trt bullshit

TNTwise · May 12, 2024 · 57686ed · 57686ed
1 parent f45e8fe
commit 57686ed
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/modules/CustomModelsCUDA.py b/modules/CustomModelsCUDA.py
@@ -44,6 +44,7 @@ def modelOptions(self):
     except:
         pass
     self.ui.Rife_Model.currentIndexChanged.connect(self.greyOutRealSRTimes)
+    self.greyOutRealSRTimes()
     self.ui.Rife_Times.setCurrentIndex(0)
     self.ui.denoiseLevelLabel.hide()
     self.ui.denoiseLevelSpinBox.hide()

diff --git a/requirements-CUDA-Linux.txt b/requirements-CUDA-Linux.txt
@@ -7,3 +7,4 @@ cupy-cuda11x
 fastrlock
 https://github.com/pytorch/TensorRT/releases/download/v1.4.0/torch_tensorrt-1.4.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
 tensorrt==8.6.0
+polygraphy
diff --git a/src/torch/RenderCUDA.py b/src/torch/RenderCUDA.py
@@ -26,9 +26,10 @@
 try:
     import tensorrt
     from src.torch.rife.tensorRT import RifeTensorRT
-    from src.torch.UpscaleImageTensorRT import UniversalTensorRT
-except:
-    pass
+    from src.torch.UpscaleImageTensorRT import UpscaleTensorRT
+except Exception as e:
+    log(str(e))
+