an><span class="n">predict</span><span class="p">(</span>
            <span class="s2">&quot;sentence_classification_head&quot;</span><span class="p">,</span>
            <span class="n">tokens</span><span class="p">[:</span><span class="mi">1024</span><span class="p">],</span>
            <span class="n">return_logits</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
        <span class="p">)</span><span class="o">.</span><span class="n">squeeze</span><span class="p">(</spa