an><span class="p">),</span>
                <span class="c1"># cuda_graph_batch_size=lower_setting.cuda_graph_batch_size, # NOTE: Not sure what this is supposed to do</span>
            <span class="p">)</span>
            <span class="k">return</span> <span class="n">trt_module</span>

        <span class="k">else</span><span class="p">:</span>
            <span class="n">trt_module</span> <span class="o">=</span> <span class="n">TRTModule</span><span class="p">(</span>
                <span class="n">eng