From c19731ec58eeb11c3d7f50ecbfa884ea4679c0ce Mon Sep 17 00:00:00 2001 From: huchukato Date: Fri, 27 Feb 2026 02:30:44 +0100 Subject: [PATCH] Fix AttributeError by initializing inputs and outputs attributes in Engine constructor --- trt_utilities.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/trt_utilities.py b/trt_utilities.py index 44deefa..0846244 100644 --- a/trt_utilities.py +++ b/trt_utilities.py @@ -248,6 +248,8 @@ class Engine: self.context = None self.buffers = OrderedDict() self.tensors = OrderedDict() + self.inputs = {} # Initialize here + self.outputs = {} # Initialize here self.cuda_graph_instance = None # cuda graph self.graph = None