"p">,</span>
    <span class="n">offload_module_to_cpu</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
<span class="p">)</span>
</pre></div>
</div>
</section>
<section id="post-processing">
<h2>Post Processing<a class="headerlink" href="#post-processing" title="Permalink to this heading">¶</a></h2>
<p>Release the GPU memory occupied by the exported program and the pipe.transformer
Set the transformer in the Flux pipeline to the Torch-TRT compiled model</p>
<div class="hig