im just going back to 8.6.0

TNTwise · May 12, 2024 · 1289cf8 · 1289cf8
1 parent dc64868
commit 1289cf8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/torch/RenderCUDA.py b/src/torch/RenderCUDA.py
@@ -24,6 +24,7 @@
 except:
     pass
 try:
+    import tensorrt
     from src.torch.rife.tensorRT import RifeTensorRT
     from src.torch.UpscaleImageTensorRT import UniversalTensorRT
 except:

diff --git a/src/torch/UpscaleImageTensorRT.py b/src/torch/UpscaleImageTensorRT.py
@@ -7,7 +7,6 @@
 import logging
 # import torch_tensorrt as trt
 
-import tensorrt_libs
 import tensorrt
 from polygraphy.backend.trt import (
     TrtRunner,