="pre">scheduler</span></code> necessary
to generate an image. We load the weights in <code class="docutils literal notranslate"><span class="pre">FP16</span></code> precision using <code class="docutils literal notranslate"><span class="pre">torch_dtype</span></code> argument</p>
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">DEVICE</span> <span class="o">=</span> <span class="s2">&quot;cuda:0&quot;</span>
<span class="n">pipe</span> <span class="o">=</sp