From 1289cf84ff386e6331a744068c33dd571438ec10 Mon Sep 17 00:00:00 2001 From: TNTwise Date: Sat, 11 May 2024 19:03:13 -0500 Subject: [PATCH] im just going back to 8.6.0 --- src/torch/RenderCUDA.py | 1 + src/torch/UpscaleImageTensorRT.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/src/torch/RenderCUDA.py b/src/torch/RenderCUDA.py index d14e46b7..a3d84076 100644 --- 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 index 52cfe68a..dcd30e3f 100644 --- 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,