l notranslate"><span class="pre">vae</span></code>, <code class="docutils literal notranslate"><span class="pre">text_encoder</span></code>, <code class="docutils literal notranslate"><span class="pre">tokenizer</span></code> and <code class="docutils literal notranslate"><span class="pre">scheduler</span></code>. In this example,
we demonstrate optimizing the <code class="docutils literal notranslate"><span class="pre">transformer</span></code> component of the model (which typically consumes &gt;95% of th